diff options
Diffstat (limited to 'lib/sqlalchemy/testing/suite/test_rowcount.py')
-rw-r--r-- | lib/sqlalchemy/testing/suite/test_rowcount.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/lib/sqlalchemy/testing/suite/test_rowcount.py b/lib/sqlalchemy/testing/suite/test_rowcount.py index bb344237a..504ac13a5 100644 --- a/lib/sqlalchemy/testing/suite/test_rowcount.py +++ b/lib/sqlalchemy/testing/suite/test_rowcount.py @@ -2,7 +2,6 @@ from sqlalchemy import bindparam from sqlalchemy import Column from sqlalchemy import Integer from sqlalchemy import select -from sqlalchemy import Sequence from sqlalchemy import String from sqlalchemy import Table from sqlalchemy import testing @@ -25,7 +24,7 @@ class RowCountTest(fixtures.TablesTest): Column( "employee_id", Integer, - Sequence("employee_id_seq", optional=True), + autoincrement=False, primary_key=True, ), Column("name", String(50)), @@ -49,7 +48,10 @@ class RowCountTest(fixtures.TablesTest): employees_table = cls.tables.employees connection.execute( employees_table.insert(), - [{"name": n, "department": d} for n, d in data], + [ + {"employee_id": i, "name": n, "department": d} + for i, (n, d) in enumerate(data) + ], ) def test_basic(self, connection): |