1#!/usr/bin/env python
2
3import unittest
4
5from idna.intranges import intranges_from_list, intranges_contain, _encode_range
6
7
8class IntrangeTests(unittest.TestCase):
9
10    def test_ranging(self):
11        self.assertEqual(
12            intranges_from_list(list(range(293, 499)) + list(range(4888, 9876))),
13            (_encode_range(293, 499), _encode_range(4888, 9876),)
14        )
15
16    def test_ranging_2(self):
17        self.assertEqual(
18            intranges_from_list([111]),
19            (_encode_range(111, 112),)
20        )
21
22    def test_skips(self):
23        self.assertEqual(
24            intranges_from_list([0, 2, 4, 6, 9, 10, 11, 13, 15,]),
25            (
26                _encode_range(0, 1),
27                _encode_range(2, 3),
28                _encode_range(4, 5),
29                _encode_range(6, 7),
30                _encode_range(9, 12),
31                _encode_range(13, 14),
32                _encode_range(15, 16),
33            )
34        )
35
36    def test_empty_range(self):
37        self.assertEqual(
38            intranges_from_list([]),
39            ()
40        )
41
42
43class IntrangeContainsTests(unittest.TestCase):
44
45    def _test_containment(self, ints, disjoint_ints):
46        ranges = intranges_from_list(ints)
47        for int_ in ints:
48            assert intranges_contain(int_, ranges)
49        for int_ in disjoint_ints:
50            assert not intranges_contain(int_, ranges)
51
52    def test_simple(self):
53        self._test_containment(range(10, 20), [2, 3, 68, 3893])
54
55    def test_skips(self):
56        self._test_containment(
57            [0, 2, 4, 6, 9, 10, 11, 13, 15,],
58            [-1, 1, 3, 5, 7, 4898]
59        )
60
61    def test_singleton(self):
62        self._test_containment([111], [110, 112])
63
64    def test_empty(self):
65        self._test_containment([], range(100))
66