lockdrop/calculations.py

@@ -179,7 +179,7 @@ def generate_test_output(final_data):
             f"{year}_years": int(final_galaxy_allocations[year] * Decimal('1e8'))
             for year in range(1, 6)
         },
-        "total_allocation_sz": int(total_allocation * Decimal('1e8'))
+        "total": int(total_allocation * Decimal('1e8'))
     }
 
     return output

tests/base_test.py

@@ -31,7 +31,8 @@ class BaseAllocationTest(unittest.TestCase):
         with open(f'{cls.generated_dir}/generated-participants.json', 'r') as f:
             cls.participants = json.load(f)
         with open('lockdrop_allocations_notebook.json', 'r') as f:
-            cls.notebook_allocations = json.load(f)
+            notebook_data = json.load(f)
+            cls.notebook_allocations = notebook_data['allocations']
 
         cls.points_by_duration = cls._get_first_points()