def test_can_return_model_as_best_assembly(self, mock_best): mock_best.return_value = None f = File() f._models = "ABCDEF" model = f.generate_best_assembly() self.assertEqual(model, "A")
def test_can_get_model(self): file_ = File() file_._models = ["1", "2"] self.assertIs(file_.model, file_._models[0])