diff options
-rw-r--r-- | selftest/tests/test_run.py | 2 | ||||
-rw-r--r-- | selftest/tests/test_testlist.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/selftest/tests/test_run.py b/selftest/tests/test_run.py index 829f488cf8..0a5a0ff52e 100644 --- a/selftest/tests/test_run.py +++ b/selftest/tests/test_run.py @@ -25,6 +25,7 @@ from samba.subunit import ( PROGRESS_PUSH, PROGRESS_POP, ) +from samba.tests import TestCase import tempfile from selftest.run import ( @@ -36,7 +37,6 @@ from selftest.run import ( run_testsuite_command, ) -from unittest import TestCase class ExpandEnvironmentStringsTests(TestCase): diff --git a/selftest/tests/test_testlist.py b/selftest/tests/test_testlist.py index c0bbebfc14..d82b00cefd 100644 --- a/selftest/tests/test_testlist.py +++ b/selftest/tests/test_testlist.py @@ -22,7 +22,7 @@ import os import tempfile -from unittest import TestCase +from samba.tests import TestCase from selftest.testlist import ( RestrictedTestManager, |