29 lines
1009 B
Python
29 lines
1009 B
Python
|
import sys
|
||
|
|
import os
|
||
|
|
sys.path.append('..')
|
||
|
|
import unittest
|
||
|
|
import timeit
|
||
|
|
|
||
|
|
import mimcvdf
|
||
|
|
|
||
|
|
class TestVDF(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_both(self):
|
||
|
|
h = mimcvdf.vdf_create(b"test")
|
||
|
|
self.assertTrue(mimcvdf.vdf_verify(b"test", h))
|
||
|
|
|
||
|
|
def test_invalid_both(self):
|
||
|
|
h = mimcvdf.vdf_create(b"test")
|
||
|
|
self.assertFalse(mimcvdf.vdf_verify(b"invalid", h))
|
||
|
|
|
||
|
|
def test_above_zero_rounds(self):
|
||
|
|
self.assertRaises(AssertionError, mimcvdf.vdf_create, b"test", 0)
|
||
|
|
self.assertRaises(AssertionError, mimcvdf.vdf_create, b"test", -1)
|
||
|
|
self.assertRaises(AssertionError, mimcvdf.vdf_create, b"test", -10000)
|
||
|
|
|
||
|
|
def test_profile(self):
|
||
|
|
rand = os.urandom(1000)
|
||
|
|
self.assertAlmostEqual(timeit.timeit(lambda: mimcvdf.vdf_create(b"test", 1000), number=100), mimcvdf.profile_cpu_speed(1000), places=2)
|
||
|
|
self.assertAlmostEqual(timeit.timeit(lambda: mimcvdf.vdf_create(rand, 1000), number=100), mimcvdf.profile_cpu_speed(1000), places=2)
|
||
|
|
|
||
|
|
unittest.main()
|