def test_removing_non_existent_label_fails(self): node = Node("Person", name="Alice") with self.assertRaises(KeyError): node.remove_label("Employee")
def test_can_remove_label(self): node = Node("Person", "Employee", name="Alice") node.remove_label("Employee") assert node.labels() == {"Person"}