Merge pull request #8 from Cycxyz/aggregated_test

Aggregated test

Author: andreysobol

.github/ci.yml

@@ -74,7 +74,7 @@ def read_events_chunked(contract, start_block, end_block, chunk_size=10000):
     all_logs = []
     current_block = start_block
 
-    while current_block <= end_block:
+    while current_block < end_block:
         chunk_end = min(current_block + chunk_size - 1, end_block)
 
         try:

nft_events.py

@@ -74,7 +74,7 @@ def read_events_chunked(contract, start_block, end_block, chunk_size=10000):
     all_logs = []
     current_block = start_block
 
-    while current_block <= end_block:
+    while current_block < end_block:
         chunk_end = min(current_block + chunk_size - 1, end_block)
 
         try:

pilot_vault_events.py

@@ -0,0 +1,37 @@
+from test_points import load_points_sorted, DATA_DIR
+from pathlib import Path
+import json
+
+
+def load_aggregated_points_sorted():
+    aggregated_points_dir = DATA_DIR / "aggregated_points"
+    aggregated_points = sorted(
+        aggregated_points_dir.glob("*.json"), key=lambda f: int(f.stem)
+    )
+    return [json.loads(f.read_text()) for f in aggregated_points]
+
+
+class TestAggregatePoints:
+    def test_aggregate_points_borders_match_points_borders(self):
+        aggregated_points = load_aggregated_points_sorted()
+        points = load_points_sorted()
+        for i in range(len(aggregated_points)):
+            for user, data in aggregated_points[i]["points"].items():
+                assert (
+                    sum(
+                        [
+                            point["points"].get(user.lower(), 0)
+                            for point in points[: i + 1]
+                        ]
+                    )
+                    == data["cumulative_points"]
+                )
+
+            assert (
+                sum([sum(point["points"].values()) for point in points[: i + 1]])
+                == aggregated_points[i]["metadata"]["total_points_all_users"]
+            )
+            assert (
+                sum(user_points for _, user_points in points[i]["points"].items())
+                == aggregated_points[i]["metadata"]["day_points"]
+            )

test/test_aggregate_points.py

@@ -35,7 +35,7 @@ def get_onchain_balances(w3, state_data, users_data, addresses, block_key):
     block = state_data[block_key] if block_key == "end_block" else state_data[block_key] - 1
     for user, _ in users_data.items():
         contract = w3.eth.contract(
-            address=Web3.to_checksum_address(["pilot_vault"]), abi=ERC20_ABI
+            address=Web3.to_checksum_address(addresses["pilot_vault"]), abi=ERC20_ABI
         )
         balance = contract.functions.balanceOf(Web3.to_checksum_address(user)).call(
             block_identifier=block