def test_get_model_to_merge(self, test_input, get_models): class Model(object): __name__ = "" return_value = [] for v in ["one", "two", "three"]: instance = Model() instance.__name__ = v return_value.append(instance) get_models.return_value = return_value test_input.return_value = 2 model_to_deduplicate = get_model_to_deduplicate() self.assertEqual(model_to_deduplicate.__name__, "two")
def test_get_model_to_merge(self, test_input, get_models): class Model(object): __name__ = "" return_value = [] for v in ["one", "two", "three"]: instance = Model() instance.__name__ = v return_value.append(instance) get_models.return_value = return_value test_input.return_value = 2 model_to_deduplicate = get_model_to_deduplicate() self.assertEqual(model_to_deduplicate.__name__, "two")