def _create_model(self) -> InternalIdModel: return create_stub_well()
def test_convert_well(self): alchemy_model = convert_to_sqlalchemy_model(create_stub_well()) converted_model = convert_to_popo_model(alchemy_model) # type: Well self.assertEqual(converted_model.__class__, Well) self.assertEqual(converted_model.internal_id, INTERNAL_ID) self.assertEqual(converted_model.name, NAME)