22import pickle
33from typing import Union
44
5- import importlib_metadata
65import pytest
76from requests import Session
87
1110from arango .exceptions import ArangoClientError , ServerConnectionError
1211from arango .http import DefaultHTTPClient , DeflateRequestCompression
1312from arango .resolver import FallbackHostResolver , RandomHostResolver , SingleHostResolver
13+ from arango .version import __version__
1414from tests .helpers import (
1515 generate_col_name ,
1616 generate_db_name ,
@@ -23,7 +23,7 @@ def test_client_attributes(url):
2323 http_client = DefaultHTTPClient ()
2424
2525 client = ArangoClient (hosts = url , http_client = http_client )
26- assert client .version == importlib_metadata . version ( "python-arango" )
26+ assert client .version == __version__
2727 assert client .hosts == [url ]
2828
2929 assert repr (client ) == f"<ArangoClient { url } >"
@@ -38,7 +38,7 @@ def test_client_attributes(url):
3838 serializer = json .dumps ,
3939 deserializer = json .loads ,
4040 )
41- assert client .version == importlib_metadata . version ( "python-arango" )
41+ assert client .version == __version__
4242 assert client .hosts == client_hosts
4343 assert repr (client ) == client_repr
4444 assert isinstance (client ._host_resolver , FallbackHostResolver )
@@ -50,7 +50,7 @@ def test_client_attributes(url):
5050 serializer = json .dumps ,
5151 deserializer = json .loads ,
5252 )
53- assert client .version == importlib_metadata . version ( "python-arango" )
53+ assert client .version == __version__
5454 assert client .hosts == client_hosts
5555 assert repr (client ) == client_repr
5656 assert isinstance (client ._host_resolver , RandomHostResolver )
0 commit comments