def get_secondary_vertex_label(self, entry):
label_weight = LGLReader.get_vertex_label_and_weight(entry)
if not label_weight:
raise ValueError
if isinstance(label_weight, list):
secondary_label = label_weight[0]
self.edge_weights.InsertNextValue(label_weight[1])
else:
secondary_label = label_weight
self.edge_weights.InsertNextValue(0.5)
return secondary_label
def test_given_entry_with_weigth_returns_label_weight_list(self):
self.assertEqual(['vertex', 42.1], LGLReader.get_vertex_label_and_weight('vertex 42.1'))
def test_given_entry_with_incorrect_etry_exception_raises(self):
try:
LGLReader.get_vertex_label_and_weight('vertex d')
self.assertTrue(False)
except ValueError:
self.assertTrue(True)
def test_given_entry_returns_label(self):
self.assertEqual('vertex', LGLReader.get_vertex_label_and_weight('vertex'))
def test_given_incorrect_entry_returns_none(self):
self.assertEqual(None, LGLReader.get_vertex_label_and_weight(''))
self.assertEqual(None, LGLReader.get_vertex_label_and_weight(' '))
