1212import pynguin .ga .computations as ff
1313import pynguin .ga .testcasechromosome as tcc
1414import pynguin .ga .testsuitechromosome as tsc
15+ from pynguin .analyses .generator import HeuristicGeneratorFitnessFunction
1516from pynguin .analyses .module import generate_test_cluster
1617from pynguin .analyses .typesystem import Instance
1718from pynguin .instrumentation .tracer import ExecutionTrace , SubjectProperties
@@ -181,7 +182,7 @@ def test_heuristic_generator_fitness_function(name, expected_fitness):
181182 cluster = generate_test_cluster ("tests.fixtures.examples.constructors" )
182183 type_system = cluster .type_system
183184 base_type = Instance (type_system .find_type_info ("tests.fixtures.examples.constructors.Base" ))
184- generator_ff = ff . HeuristicGeneratorFitnessFunction (type_system = type_system )
185+ generator_ff = HeuristicGeneratorFitnessFunction (type_system = type_system )
185186 generators = get_all_generators (cluster )
186187 assert generator_ff .compute_fitness (base_type , generators [name ]) == expected_fitness
187188
@@ -190,7 +191,7 @@ def test_heuristic_generator_fitness_function_not_connected():
190191 cluster = generate_test_cluster ("tests.fixtures.examples.constructors" )
191192 type_system = cluster .type_system
192193 base_type = Instance (type_system .find_type_info ("tests.fixtures.examples.constructors.Base" ))
193- generator_ff = ff . HeuristicGeneratorFitnessFunction (type_system = type_system )
194+ generator_ff = HeuristicGeneratorFitnessFunction (type_system = type_system )
194195 generators = get_all_generators (cluster )
195196 assert generator_ff .compute_fitness (
196197 base_type , generators ["tests.fixtures.examples.constructors.Base2" ]
@@ -199,7 +200,7 @@ def test_heuristic_generator_fitness_function_not_connected():
199200
200201def test_heuristic_generator_fitness_function_is_minimisation ():
201202 mock = MagicMock ()
202- func = ff . HeuristicGeneratorFitnessFunction (mock )
203+ func = HeuristicGeneratorFitnessFunction (mock )
203204 assert not func .is_maximisation_function ()
204205
205206
0 commit comments