|
20 | 20 | importtime |
21 | 21 | importunittest |
22 | 22 |
|
23 | | -fromdistutils.versionimportLooseVersion |
| 23 | +frompackaging.versionimportVersion |
24 | 24 | fromtestgresimportget_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 | | -ifversion<LooseVersion('9.6.0'): |
| 451 | +ifversion<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 | withnode.connect()ascon: |
487 | 487 | con.execute('set max_parallel_workers_per_gather = 2') |
488 | | -ifversion>=LooseVersion('10'): |
| 488 | +ifversion>=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 | | -ifversion<LooseVersion('14'): |
| 1048 | +ifversion<Version('14'): |
1049 | 1049 | expected_format=''' |
1050 | 1050 | { |
1051 | 1051 | "Plans": [ |
|