chore: remove dependency on sure in code

Author: martinkersner

tests/test_api.py

@@ -28,23 +28,18 @@ def test_API_initial():
     """Tests the API initialization without any parameters."""
     client = API()
 
-    client.should.be.a(API)
-    client.api_key.should.be.none
-    client.timeout.should.equal(10)
-    client.show_limit_usage.should.be.false
-    client.show_header.should.be.false
-    client.session.should.be.a(requests.Session)
-    client.session.headers.should.have.key("Content-Type").which.should.equal(
-        "application/json;charset=utf-8"
+    assert isinstance(client, API)
+    assert client.api_key is None
+    assert client.timeout == 10
+    assert client.show_limit_usage is False
+    assert client.show_header is False
+    assert isinstance(client.session, requests.Session)
+    assert (
+        client.session.headers.get("Content-Type") == "application/json;charset=utf-8"
     )
-    client.session.headers.should.have.key("User-Agent").which.should.equal(
-        "datamaxi/" + __version__
-    )
-    client.session.headers.should.have.key("X-DTMX-APIKEY").which.should.be.equal(
-        "None"
-    )
-
-    client._logger.should.be(logging.getLogger("datamaxi.api"))
+    assert client.session.headers.get("User-Agent") == "datamaxi/" + __version__
+    assert client.session.headers.get("X-DTMX-APIKEY") == "None"
+    assert client._logger is logging.getLogger("datamaxi.api")
 
 
 def test_API_env_key():
@@ -53,23 +48,19 @@ def test_API_env_key():
     os.environ["DATAMAXI_API_KEY"] = api_key
     client = API()
 
-    client.should.be.a(API)
-    client.api_key.should.be.equal(api_key)
-    client.session.headers.should.have.key("X-DTMX-APIKEY").which.should.be.equal(
-        api_key
-    )
+    assert isinstance(client, API)
+    assert client.api_key == api_key
+    assert client.session.headers.get("X-DTMX-APIKEY") == api_key
 
 
 def test_API_with_explicit_key():
     """Tests the API initialization with explicit API key parameter."""
     api_key = random_str()
     client = API(api_key)
 
-    client.should.be.a(API)
-    client.api_key.should.be.equal(api_key)
-    client.session.headers.should.have.key("X-DTMX-APIKEY").which.should.be.equal(
-        api_key
-    )
+    assert isinstance(client, API)
+    assert client.api_key == api_key
+    assert client.session.headers.get("X-DTMX-APIKEY") == api_key
 
 
 def test_API_explicit_key_overrides_env():
@@ -79,10 +70,8 @@ def test_API_explicit_key_overrides_env():
     os.environ["DATAMAXI_API_KEY"] = env_key
     client = API(explicit_key)
 
-    client.api_key.should.be.equal(explicit_key)
-    client.session.headers.should.have.key("X-DTMX-APIKEY").which.should.be.equal(
-        explicit_key
-    )
+    assert client.api_key == explicit_key
+    assert client.session.headers.get("X-DTMX-APIKEY") == explicit_key
 
 
 def test_API_with_extra_parameters():
@@ -100,29 +89,29 @@ def test_API_with_extra_parameters():
         proxies=proxies,
     )
 
-    client.should.be.a(API)
-    client.api_key.should.equal(api_key)
-    client.timeout.should.equal(0.1)
-    client.base_url.should.equal(base_url)
-    client.show_limit_usage.should.be.true
-    client.show_header.should.be.true
-    client.proxies.should.equal(proxies)
-    client.session.headers.should.have.key("X-DTMX-APIKEY").which.should.equal(api_key)
+    assert isinstance(client, API)
+    assert client.api_key == api_key
+    assert client.timeout == 0.1
+    assert client.base_url == base_url
+    assert client.show_limit_usage is True
+    assert client.show_header is True
+    assert client.proxies == proxies
+    assert client.session.headers.get("X-DTMX-APIKEY") == api_key
 
 
 def test_API_with_custom_timeout():
     """Tests the API initialization with custom timeout."""
     client = API(timeout=30)
-    client.timeout.should.equal(30)
+    assert client.timeout == 30
 
 
 def test_API_with_show_limit_usage():
     """Tests the API initialization with show_limit_usage enabled."""
     client = API(show_limit_usage=True)
-    client.show_limit_usage.should.be.true
+    assert client.show_limit_usage is True
 
 
 def test_API_with_show_header():
     """Tests the API initialization with show_header enabled."""
     client = API(show_header=True)
-    client.show_header.should.be.true
+    assert client.show_header is True