#NEWLINE# @ mock . patch . object ( costs . CostHandler , 'get_all_classes' ) #NEWLINE# def test_get_cost_classes_normal( self , getcls ) : #NEWLINE# #INDENT# self . flags ( scheduler_solver_costs = [ 'FakeCost1' ] , group = 'solver_scheduler' ) #NEWLINE# getcls . return_value = [ FakeCost1 , FakeCost2 ] #NEWLINE# cost_classes = self . solver . _get_cost_classes ( ) #NEWLINE# self . assertIn ( FakeCost1 , cost_classes ) #NEWLINE# self . assertNotIn ( FakeCost2 , cost_classes )