def test_no_mock_needed(self): gp = Program(model=[]) self.assertEquals({}, gp.get_model_mock())
def test_mock_found(self): gp = Program(model=[simple, {'mock': True}]) self.assertEquals({'mock': True}, gp.get_model_mock())
def test_no_mock_found(self): """ The program raises MockNotFound when the program has no mock defined """ gp = Program(model=[simple]) self.assertRaises(MockNotFound, lambda: gp.get_model_mock())