🦍 feat: refactor test

mmsaki · Feb 11, 2025 · fba76c5 · fba76c5
1 parent 6c985ac
commit fba76c5
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/Vertex_test.py b/tests/Vertex_test.py
@@ -1,4 +1,5 @@
 import unittest
+
 from defigraph.Vertex import Vertex
 import pytest
 from web3 import Web3
@@ -26,12 +27,14 @@ def setUp(self):
         }
 
     def test_address_should_be_valid_checksum(self):
+        name = self.pool["token0"]
+        decimals = int(self.pool["token0"]["decimals"])
+        address = self.pool["token0"]["id"]  # should be checksum address
         with pytest.raises(ValueError):
-            name = self.pool["token0"]
-            decimals = int(self.pool["token0"]["decimals"])
-            address = self.pool["token0"]["id"]
             Vertex(name, decimals, address)
 
+        Vertex(name, decimals, Web3.to_checksum_address(address))
+
     def tearDown(self):
         return super().tearDown()