def test_convert_colors_valids(self):
task = GpsVisDB()
colors = {'FFFF50': 1,
'F7E8AA': 2,
'FFA550': 3,
'5A5AFF': 4,
'BEFFFF': 5,
'8CFF8C': 6,
'FF8CFF': 7,
'AADD96': 8,
'FFD3AA': 9,
'C6C699': 10,
'E5BFC6': 11,
'DADADA': 12,
'C6B5C4': 13,
'C1D1BF': 14,
'000000': 15
}
for (code, index) in colors.iteritems():
result = task.convert_colors({'id': 1, 'color': code,
'size': 'small', 'speed': 1})
self.assertEquals(result, index)
def test_convert_colors_notfound(self):
task = GpsVisDB()
with self.assertRaises(ValueError):
task.convert_colors({'id': 1, 'color': 'blue',
'size': 'small', 'speed': 1})
