|
20 | 20 | import time
|
21 | 21 | import unittest
|
22 | 22 |
|
23 |
| -from distutils.version import LooseVersion |
| 23 | +from packaging.version import Version |
24 | 24 | from testgres import get_new_node, get_pg_version, configure_testgres
|
25 | 25 |
|
26 | 26 | # set setup base logging config, it can be turned on by `use_python_logging`
|
|
58 | 58 | }
|
59 | 59 |
|
60 | 60 | logging.config.dictConfig(LOG_CONFIG)
|
61 |
| -version = LooseVersion(get_pg_version()) |
| 61 | +version = Version(get_pg_version()) |
62 | 62 |
|
63 | 63 |
|
64 | 64 | # Helper function for json equality
|
@@ -448,7 +448,7 @@ def test_parallel_nodes(self):
|
448 | 448 |
|
449 | 449 | # Check version of postgres server
|
450 | 450 | # If version < 9.6 skip all tests for parallel queries
|
451 |
| - if version < LooseVersion('9.6.0'): |
| 451 | + if version < Version('9.6.0'): |
452 | 452 | return
|
453 | 453 |
|
454 | 454 | # Prepare test database
|
@@ -485,7 +485,7 @@ def test_parallel_nodes(self):
|
485 | 485 | # Test parallel select
|
486 | 486 | with node.connect() as con:
|
487 | 487 | con.execute('set max_parallel_workers_per_gather = 2')
|
488 |
| - if version >= LooseVersion('10'): |
| 488 | + if version >= Version('10'): |
489 | 489 | con.execute('set min_parallel_table_scan_size = 0')
|
490 | 490 | else:
|
491 | 491 | con.execute('set min_parallel_relation_size = 0')
|
@@ -1045,7 +1045,7 @@ def test_update_node_plan1(self):
|
1045 | 1045 | self.assertEqual(len(plan["Target Tables"]), 11)
|
1046 | 1046 |
|
1047 | 1047 | # Plan was seriously changed in vanilla since v14
|
1048 |
| - if version < LooseVersion('14'): |
| 1048 | + if version < Version('14'): |
1049 | 1049 | expected_format = '''
|
1050 | 1050 | {
|
1051 | 1051 | "Plans": [
|
|
0 commit comments