Update notebook results loading in tests

This commit is contained in:
Prathamesh Musale 2025-08-13 17:06:53 +05:30
parent c107afc4b3
commit c8c5b4bd15
2 changed files with 3 additions and 2 deletions

View File

@ -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

View File

@ -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()