def test_unicode_full(self):
"""Test __unicode__() method for Node with complete record."""
node = Node(self.record, set(), set())
nodestr = node.__unicode__()
nodestrexpt = u"Carl Friedrich Gau\xdf \\nUniversit\xe4t Helmstedt \
(1799)"
self.assertEqual(nodestr, nodestrexpt)
def test_equal(self):
"""Test comparison method for Nodes with identical records."""
record2 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt",
1799, 18231)
node1 = Node(self.record, set(), set())
node2 = Node(record2, set(), set())
self.assertTrue(node1 == node2)
def test_unicode_no_inst_no_id(self):
"""Test __unicode__() method for Node containing record without
institution or year."""
record = Record(self.record.name, None, None, 18231)
node = Node(record, set(), set())
nodestr = node.__unicode__()
nodestrexpt = u"Carl Friedrich Gau\xdf"
self.assertEqual(nodestr, nodestrexpt)
def test_unequal(self):
"""Test comparison method for Nodes with different records."""
record2 = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
node1 = Node(self.record, set(), set())
node2 = Node(record2, set(), set())
self.assertTrue(node1 < node2)
self.assertTrue(node1 != node2)
self.assertTrue(node2 > node1)
def test_unicode_no_year(self):
"""
Test __unicode__() method for Node containing record without year.
"""
record = Record(self.record.name, self.record.institution, None, 18231)
node = Node(record, set(), set())
nodestr = node.__unicode__()
nodestrexpt = u"Carl Friedrich Gau\xdf \\nUniversit\xe4t Helmstedt"
self.assertEqual(nodestr, nodestrexpt)
def test_add_node_object(self):
"""Test the add_node_object() method."""
record = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
node = Node(record, set(), set())
self.graph1.add_node_object(node)
self.assertEqual(set([38586, 18231]), self.graph1.get_node_list())
self.assertEqual(self.graph1.seeds, set([18231]))
def test_str_no_inst_no_id(self):
"""Test __stre__() method for Node containing record without
institution or year."""
record = Record(self.record.name, None, None, 18231)
node = Node(record, set(), set())
nodestr = str(node)
nodestrexpt = "Carl Friedrich Gau\xdf"
self.assertEqual(nodestr, nodestrexpt)
def test_str_full(self):
"""Test __str__() method for Node with complete record."""
node = Node(self.record, set(), set())
nodestr = str(node)
nodestrexpt = "Carl Friedrich Gau\xdf \\nUniversit\xe4t Helmstedt \
(1799)"
self.assertEqual(nodestr, nodestrexpt)
def test_str_no_year(self):
"""
Test __str__() method for Node containing record without year.
"""
record = Record(self.record.name, self.record.institution, None, 18231)
node = Node(record, set(), set())
nodestr = str(node)
nodestrexpt = "Carl Friedrich Gau\xdf \\nUniversit\xe4t Helmstedt"
self.assertEqual(nodestr, nodestrexpt)
def setUp(self):
self.record1 = Record("Carl Friedrich Gau\xdf",
u"Universit\xe4t Helmstedt", 1799, 18231)
self.node1 = Node(self.record1, set(), set())
self.graph1 = Graph(set([self.node1]))
def test_get_id(self):
"""Test the get_id() method."""
node = Node(self.record, set(), set())
self.assertEqual(node.get_id(), 18231)
def test_add_ancestor(self):
"""Test the add_ancestor() method."""
node = Node(self.record, set(), set())
node.add_ancestor(5)
self.assertEqual(node.ancestors, set([5]))
def test_get_id(self):
"""Test the get_id() method."""
node = Node(self.record, set(), set())
self.assertEqual(node.get_id(), 18231)
def test_add_ancestor_bad_type(self):
"""Test the add_ancestor() method for a case where the parameter type
is incorrect."""
node = Node(self.record, set(), set())
self.assertRaises(TypeError, node.add_ancestor, "5")
def test_add_ancestor(self):
"""Test the add_ancestor() method."""
node = Node(self.record, set(), set())
node.add_ancestor(5)
self.assertEqual(node.ancestors, set([5]))
def test_init(self):
"""Test the constructor."""
node = Node(self.record, set(), set())
self.assertEqual(node.record, self.record)
self.assertEqual(node.ancestors, set())
self.assertEqual(node.descendants, set())
