aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/idna/py3/tests/test_intranges.py
blob: 951f75dbba6344da6f769bf835545745bfa93fb6 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
#!/usr/bin/env python

import unittest

from idna.intranges import _encode_range, intranges_contain, intranges_from_list


class IntrangeTests(unittest.TestCase):
    def test_ranging(self):
        self.assertEqual(
            intranges_from_list(list(range(293, 499)) + list(range(4888, 9876))),
            (
                _encode_range(293, 499),
                _encode_range(4888, 9876),
            ),
        )

    def test_ranging_2(self):
        self.assertEqual(intranges_from_list([111]), (_encode_range(111, 112),))

    def test_skips(self):
        self.assertEqual(
            intranges_from_list(
                [
                    0,
                    2,
                    4,
                    6,
                    9,
                    10,
                    11,
                    13,
                    15,
                ]
            ),
            (
                _encode_range(0, 1),
                _encode_range(2, 3),
                _encode_range(4, 5),
                _encode_range(6, 7),
                _encode_range(9, 12),
                _encode_range(13, 14),
                _encode_range(15, 16),
            ),
        )

    def test_empty_range(self):
        self.assertEqual(intranges_from_list([]), ())


class IntrangeContainsTests(unittest.TestCase):
    def _test_containment(self, ints, disjoint_ints):
        ranges = intranges_from_list(ints)
        for int_ in ints:
            assert intranges_contain(int_, ranges)
        for int_ in disjoint_ints:
            assert not intranges_contain(int_, ranges)

    def test_simple(self):
        self._test_containment(range(10, 20), [2, 3, 68, 3893])

    def test_skips(self):
        self._test_containment(
            [
                0,
                2,
                4,
                6,
                9,
                10,
                11,
                13,
                15,
            ],
            [-1, 1, 3, 5, 7, 4898],
        )

    def test_singleton(self):
        self._test_containment([111], [110, 112])

    def test_empty(self):
        self._test_containment([], range(100))