class NodeTest(unittest.TestCase): def setUp(self): self.node = Node() def test_is_leaf_node(self): """ a leaf node has no children of its own """ self.node = SuffixNode("C") self.assertEqual(self.node.is_leaf_node(), True) self.node.add_child("C") self.assertEqual(self.node.is_leaf_node(), False) def test_add_child(self): self.node.add_child("T") self.assertEqual( len(self.node.children), 1)
def test_is_leaf_node(self): """ a leaf node has no children of its own """ self.node = SuffixNode("C") self.assertEqual(self.node.is_leaf_node(), True) self.node.add_child("C") self.assertEqual(self.node.is_leaf_node(), False)
def test_get_value_leaf_node(self): self.node = SuffixNode("A") self.assertEqual(self.node.is_leaf_node(), True) self.assertEqual(self.node.get_value(), "A" + self.endchar)
def setUp(self): self.node = Node()
def test_get_value(self): """ if not a leaf node, no term character in value """ self.node = SuffixNode("A") self.node.add_child("T") self.assertEqual(self.node.get_value(), "A")
def test_set_value_overrides_initial_value(self): self.node = SuffixNode("A") self.assertEqual(self.node.get_value(), "A" + self.endchar) self.node.set_value("C") self.assertEqual(self.node.get_value(), "C" + self.endchar)
def test_set_value(self): self.node = SuffixNode("A") self.node.set_value("T") self.assertEqual(self.node.get_value(), "T" + self.endchar)
def test_initialize_with_value(self): self.node = SuffixNode("G") self.assertEqual(self.node.get_value(), "G" + self.endchar)
def test_get_endchar(self): self.assertEqual(SuffixNode.get_endchar(), "$")
def setUp(self): self.node = SuffixNode("A") self.endchar = SuffixNode.get_endchar()
class SuffixNodeTest(unittest.TestCase): def setUp(self): self.node = SuffixNode("A") self.endchar = SuffixNode.get_endchar() def test_get_endchar(self): self.assertEqual(SuffixNode.get_endchar(), "$") def test_initialize_with_value(self): self.node = SuffixNode("G") self.assertEqual(self.node.get_value(), "G" + self.endchar) def test_set_value(self): self.node = SuffixNode("A") self.node.set_value("T") self.assertEqual(self.node.get_value(), "T" + self.endchar) def test_set_value_overrides_initial_value(self): self.node = SuffixNode("A") self.assertEqual(self.node.get_value(), "A" + self.endchar) self.node.set_value("C") self.assertEqual(self.node.get_value(), "C" + self.endchar) def test_set_value_with_invalid_value(self): self.assertRaises(TypeError, self.node.set_value, {'py': 'dict'}) def test_get_value(self): """ if not a leaf node, no term character in value """ self.node = SuffixNode("A") self.node.add_child("T") self.assertEqual(self.node.get_value(), "A") def test_get_value_leaf_node(self): self.node = SuffixNode("A") self.assertEqual(self.node.is_leaf_node(), True) self.assertEqual(self.node.get_value(), "A" + self.endchar)