Scaled back implementation to standard fixed limits. Implemented all core reading...

[~jspiros/python-ebml.git] / ebml / tests / test_core.py

diff --git a/ebml/tests/test_core.py b/ebml/tests/test_core.py
index bed71f3..54a46a7 100644 (file)
--- a/ebml/tests/test_core.py
+++ b/ebml/tests/test_core.py
@@ -7,17 +7,15 @@ from random import randint
 from ..core import *
 
 
-class ElementSizeTestsBase(object):
-       def assert_roundtrip(self, value, min_length=1, max_length=EBMLMaxSizeLength):
-               encoded = encode_element_size(value, min_length=min_length, max_length=max_length)
+class ElementSizeTests(unittest.TestCase):
+       def assert_roundtrip(self, value, length=None):
+               encoded = encode_element_size(value, length=length)
                encoded_stream = StringIO(encoded)
-               self.assertEqual(value, read_element_size(encoded_stream, max_length=max_length)[0])
-
-
-class ElementSizeTests(unittest.TestCase, ElementSizeTestsBase):
-       def test_undefined(self):
+               self.assertEqual(value, read_element_size(encoded_stream)[0])
+       
+       def test_unknown(self):
                for length in xrange(1, 9):
-                       self.assert_roundtrip(None, min_length=length)
+                       self.assert_roundtrip(None, length=length)
        
        def test_base_10(self):
                for value in (10**exp for exp in xrange(1, 16)):
@@ -37,27 +35,29 @@ class ElementSizeTests(unittest.TestCase, ElementSizeTestsBase):
                        self.assert_roundtrip(value)
 
 
-class LargeElementSizeTests(unittest.TestCase, ElementSizeTestsBase): # tests values that WILL be longer than 8 bytes (EBMLMaxSizeLength)
-       def test_base_10(self):
-               for value in (10**exp for exp in xrange(17, 300)):
-                       self.assert_roundtrip(value, max_length=1024)
-       
-       def test_base_2(self):
-               for value in (2**exp for exp in xrange(56, 1024)):
-                       self.assert_roundtrip(value, max_length=1024)
+class ElementIDTests(unittest.TestCase):
+       ebml_ids = (
+               0x1a45dfa3,
+               0x4286,
+               0x42f7,
+               0x42f2,
+               0x42f3,
+               0x4282,
+               0x4287,
+               0x4285,
+               0xbf,
+               0xec
+       )
        
-       def test_max_base_2(self):
-               for value in ((2**exp) - 2 for exp in xrange(57, 1024)):
-                       self.assert_roundtrip(value, max_length=1024)
+       def assert_roundtrip(self, value):
+               encoded = encode_element_id(value)
+               encoded_stream = StringIO(encoded)
+               self.assertEqual(value, read_element_id(encoded_stream)[0])
        
-       def test_random(self):
-               for value in (randint(2**56 - 1, 2**10240) for i in xrange(0, 200)):
-                       self.assert_roundtrip(value, max_length=10240)
-
+       def test_ebml_ids(self):
+               for id_ in self.ebml_ids:
+                       self.assert_roundtrip(id_)
 
-# class ElementIDTests(unittest.TestCase):
-#      def assert_roundtrip(self, value, max_length=EBMLMaxIDLength):
-#              encoded = encode_element_id(value, max_length=max_length)
 
 if __name__ == '__main__':
        unittest.main()
\ No newline at end of file