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