test: fix distribution test

lidofinance · Sep 2, 2024 · 2f9b09e · 2f9b09e
1 parent d061582
commit 2f9b09e
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/modules/csm/test_csm_module.py b/tests/modules/csm/test_csm_module.py
@@ -149,6 +149,8 @@ def test_calculate_distribution(module: CSOracle, csm: CSM):
         NodeOperatorId(5),
         NodeOperatorId(6),
         NodeOperatorId(7),
+        NodeOperatorId(8),
+        NodeOperatorId(9),
     )
 
     assert not log.operators[NodeOperatorId(1)].stuck
@@ -162,11 +164,11 @@ def test_calculate_distribution(module: CSOracle, csm: CSM):
     assert 6 in log.operators[NodeOperatorId(5)].validators
     assert 7 in log.operators[NodeOperatorId(6)].validators
 
-    assert log.operators[NodeOperatorId(0)].distributed == 625
-    assert log.operators[NodeOperatorId(1)].distributed == 3125
+    assert log.operators[NodeOperatorId(0)].distributed == 476
+    assert log.operators[NodeOperatorId(1)].distributed == 2380
     assert log.operators[NodeOperatorId(2)].distributed == 0
-    assert log.operators[NodeOperatorId(3)].distributed == 3125
-    assert log.operators[NodeOperatorId(6)].distributed == 3125
+    assert log.operators[NodeOperatorId(3)].distributed == 2380
+    assert log.operators[NodeOperatorId(6)].distributed == 2380
 
     assert log.frame == (100, 500)
     assert log.threshold == module.state.get_network_aggr().perf - 0.05