Harald Welte | d0519e0 | 2022-02-11 18:05:48 +0100 | [diff] [blame^] | 1 | #!/usr/bin/env python3 |
| 2 | |
| 3 | import unittest |
| 4 | from pySim.construct import GreedyInteger |
| 5 | |
| 6 | tests = [ |
| 7 | ( b'\x80', 0x80 ), |
| 8 | ( b'\x80\x01', 0x8001 ), |
| 9 | ( b'\x80\x00\x01', 0x800001 ), |
| 10 | ( b'\x80\x23\x42\x01', 0x80234201 ), |
| 11 | ] |
| 12 | |
| 13 | class TestGreedyInt(unittest.TestCase): |
| 14 | def test_GreedyInt_decoder(self): |
| 15 | gi = GreedyInteger() |
| 16 | for t in tests: |
| 17 | self.assertEqual(gi.parse(t[0]), t[1]) |
| 18 | def test_GreedyInt_encoder(self): |
| 19 | gi = GreedyInteger() |
| 20 | for t in tests: |
| 21 | self.assertEqual(t[0], gi.build(t[1])) |
| 22 | pass |
| 23 | |
| 24 | |
| 25 | if __name__ == "__main__": |
| 26 | unittest.main() |