unittest is not used

dmitry-lipetsk · dmitry-lipetsk · commit df199b92d7d9 · 2025-02-24T00:33:34.000+03:00
diff --git a/tests/test_simple.py b/tests/test_simple.py
@@ -8,7 +8,6 @@
 import testgres
 import time
 import six
-import unittest
 import pytest
 import psutil
 import platform
@@ -107,7 +106,7 @@ def removing(f):
             rmtree(f, ignore_errors=True)
 
 
-class TestgresTests(unittest.TestCase):
+class TestgresTests:
     def test_node_repr(self):
         with get_new_node() as node:
             pattern = r"PostgresNode\(name='.+', port=.+, base_dir='.+'\)"
@@ -1342,23 +1341,3 @@ def helper__skip_test_if_pg_version_is_ge(version: str):
         if pg_version_ge(version):
             pytest.skip('requires <{0}'.format(version))
 
-
-if __name__ == '__main__':
-    if os.environ.get('ALT_CONFIG'):
-        suite = unittest.TestSuite()
-
-        # Small subset of tests for alternative configs (PG_BIN or PG_CONFIG)
-        suite.addTest(TestgresTests('test_pg_config'))
-        suite.addTest(TestgresTests('test_pg_ctl'))
-        suite.addTest(TestgresTests('test_psql'))
-        suite.addTest(TestgresTests('test_replicate'))
-
-        print('Running tests for alternative config:')
-        for t in suite:
-            print(t)
-        print()
-
-        runner = unittest.TextTestRunner()
-        runner.run(suite)
-    else:
-        unittest.main()
diff --git a/tests/test_simple_remote.py b/tests/test_simple_remote.py
@@ -9,7 +9,6 @@
 import testgres
 import time
 import six
-import unittest
 import pytest
 import psutil
 
@@ -95,7 +94,7 @@ def removing(f):
             os_ops.rmdirs(f, ignore_errors=True)
 
 
-class TestgresRemoteTests(unittest.TestCase):
+class TestgresRemoteTests:
     def test_node_repr(self):
         with get_remote_node(conn_params=conn_params) as node:
             pattern = r"PostgresNode\(name='.+', port=.+, base_dir='.+'\)"
@@ -1106,23 +1105,3 @@ def helper__skip_test_if_pg_version_is_ge(version: str):
         if pg_version_ge(version):
             pytest.skip('requires <{0}'.format(version))
 
-
-if __name__ == '__main__':
-    if os_ops.environ('ALT_CONFIG'):
-        suite = unittest.TestSuite()
-
-        # Small subset of tests for alternative configs (PG_BIN or PG_CONFIG)
-        suite.addTest(TestgresRemoteTests('test_pg_config'))
-        suite.addTest(TestgresRemoteTests('test_pg_ctl'))
-        suite.addTest(TestgresRemoteTests('test_psql'))
-        suite.addTest(TestgresRemoteTests('test_replicate'))
-
-        print('Running tests for alternative config:')
-        for t in suite:
-            print(t)
-        print()
-
-        runner = unittest.TextTestRunner()
-        runner.run(suite)
-    else:
-        unittest.main()
