diff --git a/tests/test_anonvdf.py b/tests/test_anonvdf.py index eef7b5c..52379a4 100644 --- a/tests/test_anonvdf.py +++ b/tests/test_anonvdf.py @@ -9,12 +9,14 @@ from math import floor class TestAnonVDF(unittest.TestCase): def test_vdf_create(self): test_data = kasten.generator.pack.pack(b"test", "tst", 0) - test_vdf = mimcvdf.vdf_create(test_data, 1000) + test_vdf = mimcvdf.vdf_create(test_data, 1000, dec=True) + generated = anonvdf.AnonVDFGenerator.generate(test_data, rounds=1000) self.assertEqual( - anonvdf.AnonVDFGenerator.generate(test_data, rounds=1000).get_packed(), + generated.get_packed(), test_data ) + self.assertEqual(generated.id, int(test_vdf).to_bytes(64, byteorder="big")) def test_vdf_block_validate_ok_time(self): rds = 8000