Skip to content

Commit 92b69d8

Browse files
committed
Replaced deprecated python LooseVersion function
1 parent d671416 commit 92b69d8

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

Diff for: tests/python/partitioning_test.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
import time
2121
import unittest
2222

23-
from distutils.version import LooseVersion
23+
from packaging.version import Version
2424
from testgres import get_new_node, get_pg_version, configure_testgres
2525

2626
# set setup base logging config, it can be turned on by `use_python_logging`
@@ -58,7 +58,7 @@
5858
}
5959

6060
logging.config.dictConfig(LOG_CONFIG)
61-
version = LooseVersion(get_pg_version())
61+
version = Version(get_pg_version())
6262

6363

6464
# Helper function for json equality
@@ -448,7 +448,7 @@ def test_parallel_nodes(self):
448448

449449
# Check version of postgres server
450450
# If version < 9.6 skip all tests for parallel queries
451-
if version < LooseVersion('9.6.0'):
451+
if version < Version('9.6.0'):
452452
return
453453

454454
# Prepare test database
@@ -485,7 +485,7 @@ def test_parallel_nodes(self):
485485
# Test parallel select
486486
with node.connect() as con:
487487
con.execute('set max_parallel_workers_per_gather = 2')
488-
if version >= LooseVersion('10'):
488+
if version >= Version('10'):
489489
con.execute('set min_parallel_table_scan_size = 0')
490490
else:
491491
con.execute('set min_parallel_relation_size = 0')
@@ -1045,7 +1045,7 @@ def test_update_node_plan1(self):
10451045
self.assertEqual(len(plan["Target Tables"]), 11)
10461046

10471047
# Plan was seriously changed in vanilla since v14
1048-
if version < LooseVersion('14'):
1048+
if version < Version('14'):
10491049
expected_format = '''
10501050
{
10511051
"Plans": [

0 commit comments

Comments
 (0)