summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/testing/suite/test_rowcount.py
diff options
context:
space:
mode:
Diffstat (limited to 'lib/sqlalchemy/testing/suite/test_rowcount.py')
-rw-r--r--lib/sqlalchemy/testing/suite/test_rowcount.py12
1 files changed, 5 insertions, 7 deletions
diff --git a/lib/sqlalchemy/testing/suite/test_rowcount.py b/lib/sqlalchemy/testing/suite/test_rowcount.py
index ae189bc79..06945ff2a 100644
--- a/lib/sqlalchemy/testing/suite/test_rowcount.py
+++ b/lib/sqlalchemy/testing/suite/test_rowcount.py
@@ -6,7 +6,6 @@ from sqlalchemy import String
from sqlalchemy import Table
from sqlalchemy import testing
from sqlalchemy import text
-from sqlalchemy.testing import config
from sqlalchemy.testing import eq_
from sqlalchemy.testing import fixtures
@@ -33,7 +32,7 @@ class RowCountTest(fixtures.TablesTest):
)
@classmethod
- def insert_data(cls):
+ def insert_data(cls, connection):
cls.data = data = [
("Angela", "A"),
("Andrew", "A"),
@@ -47,11 +46,10 @@ class RowCountTest(fixtures.TablesTest):
]
employees_table = cls.tables.employees
- with config.db.begin() as conn:
- conn.execute(
- employees_table.insert(),
- [{"name": n, "department": d} for n, d in data],
- )
+ connection.execute(
+ employees_table.insert(),
+ [{"name": n, "department": d} for n, d in data],
+ )
def test_basic(self):
employees_table = self.tables.employees