diff options
Diffstat (limited to 'Lib/sqlite3/test/test_dbapi.py')
| -rw-r--r-- | Lib/sqlite3/test/test_dbapi.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/Lib/sqlite3/test/test_dbapi.py b/Lib/sqlite3/test/test_dbapi.py index 34cadeebfa..732e21dd3a 100644 --- a/Lib/sqlite3/test/test_dbapi.py +++ b/Lib/sqlite3/test/test_dbapi.py @@ -38,13 +38,14 @@ from test.support.os_helper import TESTFN, unlink, temp_dir # Helper for tests using TESTFN @contextlib.contextmanager -def managed_connect(*args, **kwargs): +def managed_connect(*args, in_mem=False, **kwargs): cx = sqlite.connect(*args, **kwargs) try: yield cx finally: cx.close() - unlink(TESTFN) + if not in_mem: + unlink(TESTFN) class ModuleTests(unittest.TestCase): |