lbry-sdk/lbry/tests/unit/dht/protocol/test_distance.py
2019-06-20 22:50:39 -04:00

13 lines
521 B
Python

import unittest
from lbry.dht.protocol.distance import Distance
class DistanceTests(unittest.TestCase):
def test_invalid_key_length(self):
self.assertRaises(ValueError, Distance, b'1' * 47)
self.assertRaises(ValueError, Distance, b'1' * 49)
self.assertRaises(ValueError, Distance, b'')
self.assertRaises(ValueError, Distance(b'0' * 48), b'1' * 47)
self.assertRaises(ValueError, Distance(b'0' * 48), b'1' * 49)
self.assertRaises(ValueError, Distance(b'0' * 48), b'')