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
|
# -*- coding: utf-8 -*-
from urwid_utils import Palette, PaletteEntry
from random import choice
import unittest
class TestPalette(unittest.TestCase):
def setUp(self):
pass
def test_all(self):
how_many_entries = 200
attr_names = PaletteEntry.attrs
colors = ['dark red', 'light blue', 'white', 'yellow', 'dark gray']
attr_values = ['some_palette_entry'] + colors
kwargs = dict(list(zip(attr_names, attr_values)))
some_palette_entry = PaletteEntry(**kwargs)
for name, value in zip(attr_names, attr_values):
self.assertEqual(getattr(some_palette_entry, name), value)
for index, name in enumerate(attr_names):
if index == 0 and name == 'name':
continue
setattr(some_palette_entry, name, choice(attr_values[1:]))
self.assertEqual(getattr(some_palette_entry, name), some_palette_entry[index])
some_palette_entry[index] = choice(attr_values[1:])
self.assertEqual(getattr(some_palette_entry, name), some_palette_entry[index])
entries = []
attr_values = [None] + colors
for number in range(1, how_many_entries):
name = 'p' + str(number)
palette_entry = PaletteEntry(**kwargs)
if number % 2:
palette_entry.name = name
else:
palette_entry[0] = name
entries.append((name, palette_entry))
entries = dict(entries)
palette = Palette(name='palette', **entries)
self.assertEqual(len(palette), how_many_entries - 1)
for number in range(1, how_many_entries):
palette_entry = getattr(palette, 'p' + str(number))
self.assertIn(palette_entry, palette)
|
