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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
#!/router/bin/python
import functional_general_test
from trex_stl_lib.utils import filters
from nose.tools import assert_equal
from nose.tools import assert_not_equal
from nose.tools import assert_raises
from nose.tools import assert_true, assert_false
from nose.tools import raises
class ToggleFilter_Test(functional_general_test.CGeneralFunctional_Test):
def setUp(self):
self.list_db = [1, 2, 3, 4, 5]
self.set_db = {1, 2, 3, 4, 5}
self.tuple_db = (1, 2, 3, 4, 5)
self.dict_db = {str(x): x**2
for x in range(5)}
def test_init_with_dict(self):
toggle_filter = filters.ToggleFilter(self.dict_db)
assert_equal(toggle_filter._toggle_db, set(self.dict_db.keys()))
assert_equal(toggle_filter.filter_items(), self.dict_db)
def test_init_with_list(self):
toggle_filter = filters.ToggleFilter(self.list_db)
assert_equal(toggle_filter._toggle_db, set(self.list_db))
assert_equal(toggle_filter.filter_items(), self.list_db)
def test_init_with_set(self):
toggle_filter = filters.ToggleFilter(self.set_db)
assert_equal(toggle_filter._toggle_db, self.set_db)
assert_equal(toggle_filter.filter_items(), self.set_db)
def test_init_with_tuple(self):
toggle_filter = filters.ToggleFilter(self.tuple_db)
assert_equal(toggle_filter._toggle_db, set(self.tuple_db))
assert_equal(toggle_filter.filter_items(), self.tuple_db)
@raises(TypeError)
def test_init_with_non_iterable(self):
toggle_filter = filters.ToggleFilter(15)
def test_dict_toggeling(self):
toggle_filter = filters.ToggleFilter(self.dict_db)
assert_false(toggle_filter.toggle_item("3"))
assert_equal(toggle_filter._toggle_db, {'0', '1', '2', '4'})
assert_true(toggle_filter.toggle_item("3"))
assert_equal(toggle_filter._toggle_db, {'0', '1', '2', '3', '4'})
assert_false(toggle_filter.toggle_item("2"))
assert_false(toggle_filter.toggle_item("4"))
self.dict_db.update({'5': 25, '6': 36})
assert_true(toggle_filter.toggle_item("6"))
assert_equal(toggle_filter.filter_items(), {'0': 0, '1': 1, '3': 9, '6': 36})
del self.dict_db['1']
assert_equal(toggle_filter.filter_items(), {'0': 0, '3': 9, '6': 36})
def test_dict_toggeling_negative(self):
toggle_filter = filters.ToggleFilter(self.dict_db)
assert_raises(KeyError, toggle_filter.toggle_item, "100")
def test_list_toggeling(self):
toggle_filter = filters.ToggleFilter(self.list_db)
assert_false(toggle_filter.toggle_item(3))
assert_equal(toggle_filter._toggle_db, {1, 2, 4, 5})
assert_true(toggle_filter.toggle_item(3))
assert_equal(toggle_filter._toggle_db, {1, 2, 3, 4, 5})
assert_false(toggle_filter.toggle_item(2))
assert_false(toggle_filter.toggle_item(4))
self.list_db.extend([6 ,7])
assert_true(toggle_filter.toggle_item(6))
assert_equal(toggle_filter.filter_items(), [1, 3 , 5, 6])
self.list_db.remove(1)
assert_equal(toggle_filter.filter_items(), [3, 5, 6])
def test_list_toggling_negative(self):
toggle_filter = filters.ToggleFilter(self.list_db)
assert_raises(KeyError, toggle_filter.toggle_item, 10)
def test_toggle_multiple_items(self):
toggle_filter = filters.ToggleFilter(self.list_db)
assert_false(toggle_filter.toggle_items(1, 3, 5))
assert_equal(toggle_filter._toggle_db, {2, 4})
assert_true(toggle_filter.toggle_items(1, 5))
assert_equal(toggle_filter._toggle_db, {1, 2, 4, 5})
def test_dont_show_after_init(self):
toggle_filter = filters.ToggleFilter(self.list_db, show_by_default = False)
assert_equal(toggle_filter._toggle_db, set())
assert_equal(toggle_filter.filter_items(), [])
def tearDown(self):
pass
|
