def test_given_incorrect_color_entry_returns_none(self):
self.assertEqual(None, LGLReader.get_edge_color_entry(""))
self.assertEqual(None, LGLReader.get_edge_color_entry(" "))
self.assertEqual(None, LGLReader.get_edge_color_entry("# sdf"))
self.assertEqual(None, LGLReader.get_edge_color_entry("v1 v2"))
self.assertEqual(None, LGLReader.get_edge_color_entry("v1 v2 d"))
self.assertEqual(None, LGLReader.get_edge_color_entry("v1 v2 1.0"))
self.assertEqual(None, LGLReader.get_edge_color_entry("v1 v2 1.0 d"))
self.assertEqual(None, LGLReader.get_edge_color_entry("v1 v2 1.0 0.4"))
self.assertEqual(None, LGLReader.get_edge_color_entry("v1 v2 1.0 0.4 d"))
def read_edge_colors(self):
with open(self.edge_color_filename, 'rb') as edge_colors:
color_entries = edge_colors.read()
for i, entry in enumerate(color_entries.split(os.linesep)):
if not entry:
continue
t = LGLReader.get_edge_color_entry(entry)
if not t:
print('Bad color entry at' + str(i) + ': ' + str(t))
else:
self.edge_color_tuples[t[0]] = t[1]
def test_given_correct_color_returns_v1_v2_r_g_b_tuple(self):
self.assertEqual((('v1', 'v2'), (1.0, 0.4, 0.3)),
LGLReader.get_edge_color_entry("v1 v2 1.0 0.4 0.3"))
