Fix tests for the new CDK template generator. The previous test file was not recognised as it was named unit_tests instead of unit_test. Fix this.