def test_get_pks_multiple(self): Base = declarative_base() class MyModel(Base): __tablename__ = 'blah' id = Column(Integer, primary_key=True) pk = Column(String, primary_key=True) value = Column(String) self.assertAllIn(_get_pks(MyModel), ('id', 'pk',))
def test_get_pks_single(self): Base = declarative_base() class MyModel(Base): __tablename__ = 'blah' id = Column(Integer, primary_key=True) value = Column(String) resp = _get_pks(MyModel) self.assertAllIn(('id',), resp)
def test_get_pks_single(self): Base = declarative_base() class MyModel(Base): __tablename__ = 'blah' id = Column(Integer, primary_key=True) value = Column(String) resp = _get_pks(MyModel) self.assertAllIn(('id', ), resp)
def test_get_pks_multiple(self): Base = declarative_base() class MyModel(Base): __tablename__ = 'blah' id = Column(Integer, primary_key=True) pk = Column(String, primary_key=True) value = Column(String) self.assertAllIn(_get_pks(MyModel), ( 'id', 'pk', ))