def test_check_perfect_solution(self): n = 20 w = 40 h = 40 dg = DataGenerator() some_instance_visual = dg.gen_instance_visual(n, w, h) perfect_bin_configuration = sorted(some_instance_visual, key=lambda x: (x[2][0], x[2][1])) some_instance_np_array = dg._transform_instance_visual_to_np_array(some_instance_visual) solution_checker = SolutionChecker(n, h, w) self.assertEqual( solution_checker.get_reward(np.array(perfect_bin_configuration)), 0 )