[geneagrapher] 151/226: Copying tests.py into several files in preparation to split tests.
Doug Torrance
dtorrance-guest at moszumanska.debian.org
Sat Jul 11 17:10:57 UTC 2015
This is an automated email from the git hooks/post-receive script.
dtorrance-guest pushed a commit to branch master
in repository geneagrapher.
commit cebd3735d3e1f0be5da58df0dbdae6485f2586d6
Author: David Alber <alber.david at gmail.com>
Date: Sun Oct 30 10:28:07 2011 -0700
Copying tests.py into several files in preparation to split tests.
---
tests/test_geneagrapher_methods.py | 455 +++++++++++++++++++++++++++++++++++++
tests/test_grabber_methods.py | 455 +++++++++++++++++++++++++++++++++++++
tests/test_graph_methods.py | 455 +++++++++++++++++++++++++++++++++++++
tests/test_node_methods.py | 455 +++++++++++++++++++++++++++++++++++++
tests/test_record_methods.py | 455 +++++++++++++++++++++++++++++++++++++
5 files changed, 2275 insertions(+)
diff --git a/tests/test_geneagrapher_methods.py b/tests/test_geneagrapher_methods.py
new file mode 100644
index 0000000..32d11b5
--- /dev/null
+++ b/tests/test_geneagrapher_methods.py
@@ -0,0 +1,455 @@
+import sys
+import unittest
+from geneagrapher import geneagrapher
+from geneagrapher.grabber import Grabber
+from geneagrapher.graph import DuplicateNodeError, Graph, Node, Record
+
+# Unit tests for graph-related classes.
+class TestRecordMethods(unittest.TestCase):
+ """
+ Unit tests for the Record class.
+ """
+ def test001_init(self):
+ # Test the constructor.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assertEqual(record.name, "Carl Friedrich Gauss")
+ self.assertEqual(record.institution, "Universitaet Helmstedt")
+ self.assertEqual(record.year, 1799)
+ self.assertEqual(record.id, 18231)
+
+ def test002_init_bad_name(self):
+ # Test constructor with bad 'name' parameter.
+ self.assertRaises(TypeError, Record, 1, "Universitaet Helmstedt", 1799, 18231)
+
+ def test003_init_bad_institution(self):
+ # Test constructor with bad 'institution' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss", 1, 1799, 18231)
+
+ def test004_init_bad_year(self):
+ # Test constructor with bad 'year' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss",
+ "Universitaet Helmstedt", "1799", 18231)
+
+ def test005_init_bad_id(self):
+ # Test constructor with bad 'id' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss",
+ "Universitaet Helmstedt", 1799, "18231")
+
+ def test006_cmp_equal(self):
+ # Verify two 'equal' records are compared correctly.
+ record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ record2 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record1 == record2)
+
+ def test007_cmp_unequal(self):
+ # Verify two 'unequal' records are compared correctly.
+ record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ record2 = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ self.assert_(record1 < record2)
+
+ def test008_has_institution_yes(self):
+ # Verify has_institution() method returns True when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record.has_institution())
+
+ def test009_has_institution_no(self):
+ # Verify has_institution() method returns False when the conditions are right.
+ record = Record("Carl Friedrich Gauss", None, 1799, 18231)
+ self.assert_(not record.has_institution())
+
+ def test010_has_year_yes(self):
+ # Verify has_year() method returns True when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record.has_year())
+
+ def test011_has_year_no(self):
+ # Verify has_year() method returns False when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", None, 18231)
+ self.assert_(not record.has_year())
+
+class TestNodeMethods(unittest.TestCase):
+ """
+ Unit tests for the Node class.
+ """
+ def setUp(self):
+ self.record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+
+ def test001_init(self):
+ # Test the constructor.
+ node = Node(self.record, [], [])
+ self.assertEquals(node.record, self.record)
+ self.assertEquals(node.ancestors, [])
+ self.assertEquals(node.descendants, [])
+
+ def test002_init_bad_record(self):
+ # Test the constructor for a case where the record passed is not a Record
+ # object.
+ self.assertRaises(TypeError, Node, 1, [], [])
+
+ def test003_init_bad_ancestor_list(self):
+ # Test the constructor for a case where the ancestor list is not a list.
+ self.assertRaises(TypeError, Node, self.record, 1, [])
+
+ def test003_2_init_bad_descendent_list(self):
+ # Test the constructor for a case where the descendent list is not a list.
+ self.assertRaises(TypeError, Node, self.record, [], 1)
+
+ def test004_str_full(self):
+ # Test __str__() method for Node with complete record.
+ node = Node(self.record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test005_str_no_year(self):
+ # Test __str__() method for Node containing record without year.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", None, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\nUniversitaet Helmstedt"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test006_str_no_inst(self):
+ # Test __str__() method for Node containing record without institution.
+ record = Record("Carl Friedrich Gauss", None, 1799, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\n(1799)"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test007_str_no_inst_no_id(self):
+ # Test __str__() method for Node containing record without institution
+ # or year.
+ record = Record("Carl Friedrich Gauss", None, None, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test008_cmp_equal(self):
+ # Test comparison method for Nodes with identical records.
+ record2 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ node1 = Node(self.record, [], [])
+ node2 = Node(record2, [], [])
+ self.assert_(node1 == node2)
+
+ def test009_cmp_unequal(self):
+ # Test comparison method for Nodes with different records.
+ record2 = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ node1 = Node(self.record, [], [])
+ node2 = Node(record2, [], [])
+ self.assert_(node1 < node2)
+
+ def test010_add_ancestor(self):
+ # Test the add_ancestor() method.
+ node = Node(self.record, [], [])
+ node.add_ancestor(5)
+ self.assertEquals(node.ancestors, [5])
+
+ def test011_add_ancestor_bad_type(self):
+ # Test the add_ancestor() method for a case where the parameter type is incorrect.
+ node = Node(self.record, [], [])
+ self.assertRaises(TypeError, node.add_ancestor, '5')
+
+ def test012_get_id(self):
+ node = Node(self.record, [], [])
+ self.assertEquals(node.get_id(), 18231)
+
+ def test013_set_id(self):
+ # Test the set_id() method.
+ node = Node(self.record, [], [])
+ self.assertEquals(node.get_id(), 18231)
+ node.set_id(15)
+ self.assertEquals(node.get_id(), 15)
+
+
+class TestGraphMethods(unittest.TestCase):
+ """
+ Unit tests for the Graph class.
+ """
+ def setUp(self):
+ self.record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.node1 = Node(self.record1, [], [])
+ self.graph1 = Graph([self.node1])
+
+ def test001_init_empty(self):
+ # Test the constructor.
+ graph = Graph()
+ self.assertEquals(graph.heads, None)
+
+ def test002_init(self):
+ # Test the constructor.
+ self.assert_(self.graph1.heads == [self.node1])
+ self.assertEquals(self.graph1.nodes.keys(), [18231])
+ self.assertEquals(self.graph1.nodes[18231], self.node1)
+
+ def test003_init_bad_heads(self):
+ # Test the constructor when passed a bad type for the heads parameter.
+ self.assertRaises(TypeError, Graph, 3)
+
+ def test004_has_node_true(self):
+ # Test the has_node() method for a True case.
+ self.assertEquals(self.graph1.has_node(18231), True)
+
+ def test005_has_node_false(self):
+ # Test the has_node() method for a False case.
+ self.assertEquals(self.graph1.has_node(1), False)
+
+ def test006_get_node(self):
+ # Test the get_node() method.
+ node = self.graph1.get_node(18231)
+ self.assert_(node == self.node1)
+
+ def test007_get_node_not_found(self):
+ # Test the get_node() method for a case where the node does not exist.
+ node = self.graph1.get_node(1)
+ self.assertEquals(node, None)
+
+ def test008_get_node_list(self):
+ # Test the get_node_list() method.
+ self.assertEquals(self.graph1.get_node_list(), [18231])
+
+ def test008_get_node_list_empty(self):
+ # Test the get_node_list() method for an empty graph.
+ graph = Graph()
+ self.assertEquals(graph.get_node_list(), [])
+
+ def test009_add_node(self):
+ # Test the add_node() method.
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1])
+
+ def test010_add_second_node_head(self):
+ # Test the add_node() method when adding a second node and
+ # marking it as a head node.
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [], True)
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1, self.graph1.get_node(38586)])
+
+ def test011_add_node_head(self):
+ # Test the add_node() method when no heads exist.
+ graph = Graph()
+ self.assertEquals(graph.heads, None)
+ graph.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals(graph.heads, [graph.get_node(38586)])
+
+ def test012_add_node_already_present(self):
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertRaises(DuplicateNodeError, self.graph1.add_node, "Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+
+ def test013_add_node_object(self):
+ # Test the add_node_object() method.
+ record = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ node = Node(record, [], [])
+ self.graph1.add_node_object(node)
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1])
+
+ def test014_generate_dot_file(self):
+ # Test the generate_dot_file() method.
+ dotfileexpt = """digraph genealogy {
+ graph [charset="utf-8"];
+ node [shape=plaintext];
+ edge [style=bold];
+
+ 18231 [label="Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"];
+
+}
+"""
+ dotfile = self.graph1.generate_dot_file(True, False)
+ self.assertEquals(dotfile, dotfileexpt)
+
+ def test015_generate_dot_file(self):
+ # Test the generate_dot_file() method.
+ graph = Graph()
+ graph.add_node("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231, [18230], [])
+ graph.add_node("Johann Friedrich Pfaff", "Georg-August-Universitaet Goettingen", 1786, 18230, [66476], [])
+ graph.add_node("Abraham Gotthelf Kaestner", "Universitaet Leipzig", 1739, 66476, [57670], [])
+ graph.add_node("Christian August Hausen", "Martin-Luther-Universitaet Halle-Wittenberg", 1713, 57670, [72669], [])
+ graph.add_node("Johann Christoph Wichmannshausen", "Universitaet Leipzig", 1685, 72669, [21235], [])
+ graph.add_node("Otto Mencke", "Universitaet Leipzig", 1665, 21235, [], [])
+
+ dotfileexpt = """digraph genealogy {
+ graph [charset="utf-8"];
+ node [shape=plaintext];
+ edge [style=bold];
+
+ 18231 [label="Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"];
+ 18230 [label="Johann Friedrich Pfaff \\nGeorg-August-Universitaet Goettingen (1786)"];
+ 66476 [label="Abraham Gotthelf Kaestner \\nUniversitaet Leipzig (1739)"];
+ 57670 [label="Christian August Hausen \\nMartin-Luther-Universitaet Halle-Wittenberg (1713)"];
+ 72669 [label="Johann Christoph Wichmannshausen \\nUniversitaet Leipzig (1685)"];
+ 21235 [label="Otto Mencke \\nUniversitaet Leipzig (1665)"];
+
+ 18230 -> 18231;
+ 66476 -> 18230;
+ 57670 -> 66476;
+ 72669 -> 57670;
+ 21235 -> 72669;
+}
+"""
+ dotfile = graph.generate_dot_file(True, False)
+ self.assertEquals(dotfile, dotfileexpt)
+
+class TestGrabberMethods(unittest.TestCase):
+ """
+ Unit tests for the Grabber class.
+ """
+ def setUp(self):
+ self.grabber = Grabber(18231)
+
+ def test001_init(self):
+ # Test constructor.
+ self.assertEquals(self.grabber.id, 18231)
+ self.assertEquals(self.grabber.pagestr, None)
+ self.assertEquals(self.grabber.name, None)
+ self.assertEquals(self.grabber.institution, None)
+ self.assertEquals(self.grabber.year, None)
+ self.assertEquals(self.grabber.advisors, [])
+ self.assertEquals(self.grabber.descendants, [])
+
+ def test002_get_page(self):
+ # Test get_page() method.
+ self.grabber.get_page()
+ self.assert_(self.grabber.pagestr is not None)
+ self.assert_(u"<title>The Mathematics Genealogy Project - Carl Gau\xdf</title>" in self.grabber.pagestr)
+ # Get page again and test for adverse affects.
+ self.grabber.get_page()
+ self.assert_(u"<title>The Mathematics Genealogy Project - Carl Gau\xdf</title>" in self.grabber.pagestr)
+
+ def test003_extract_info_bad(self):
+ # Verify exception thrown for bad id.
+ grabber = Grabber(999999999)
+ self.assertRaises(ValueError, grabber.extract_node_information)
+
+ def test004_extract_info_all_fields(self):
+ # Test the extract_node_information() method for a record containing all fields.
+ [name, institution, year, advisors, descendents] = self.grabber.extract_node_information()
+ self.assertEquals(name, self.grabber.name)
+ self.assertEquals(institution, self.grabber.institution)
+ self.assertEquals(year, self.grabber.year)
+ self.assertEquals(advisors, self.grabber.advisors)
+ self.assertEquals(name, u"Carl Friedrich Gau\xdf")
+ self.assertEquals(institution, u"Universit\xe4t Helmstedt")
+ self.assertEquals(year, 1799)
+ self.assertEquals(advisors, [18230])
+ self.assertEquals(descendents, [18603, 18233, 62547, 29642, 55175, 29458, 19953, 18232, 151876])
+
+ # Verify calling extract_node_information() twice does not have side effect.
+ [name, institution, year, advisors, descendents] = self.grabber.extract_node_information()
+ self.assertEquals(name, u"Carl Friedrich Gau\xdf")
+ self.assertEquals(institution, u"Universit\xe4t Helmstedt")
+ self.assertEquals(year, 1799)
+ self.assertEquals(advisors, [18230])
+ self.assertEquals(descendents, [18603, 18233, 62547, 29642, 55175, 29458, 19953, 18232, 151876])
+
+ def test005_extract_info_no_advisor(self):
+ # Test the extract_node_information() method for a record with no advisor.
+ grabber = Grabber(137717)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Valentin Alberti")
+ self.assertEquals(institution, u"Universit\xe4t Leipzig")
+ self.assertEquals(year, 1678)
+ self.assertEquals(advisors, [])
+ self.assertEquals(descendents, [127946])
+
+ def test006_extract_info_no_year(self):
+ # Test the extract_node_information() method for a record with no year.
+ # This example also has no descendents.
+ grabber = Grabber(53658)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"S. Cingolani")
+ self.assertEquals(institution, u"Scuola Normale Superiore di Pisa")
+ self.assertEquals(year, None)
+ self.assertEquals(advisors, [51261])
+ self.assertEquals(descendents, [])
+
+ def test007_extract_info_no_inst(self):
+ # Test the extract_node_information() method for a record with no institution.
+ # This test is also missing additional information already tested.
+ grabber = Grabber(52965)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Walter Mayer")
+ self.assertEquals(institution, None)
+ self.assertEquals(year, None)
+ self.assertEquals(advisors, [])
+ self.assertEquals(descendents, [52996])
+
+ # Tests for special (from my point of view) characters:
+ def test008_slash_l(self):
+ # Test the extract_node_information() method for a record
+ # containing a slash l character. Example:
+ # http://www.genealogy.math.ndsu.nodak.edu/id.php?id=7383.
+ grabber = Grabber(7383)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"W\u0142adys\u0142aw Hugo Dyonizy Steinhaus")
+ self.assertEquals(institution, u"Georg-August-Universit\xe4t G\xf6ttingen")
+ self.assertEquals(year, 1911)
+ self.assertEquals(advisors, [7298])
+
+ def test009_multiple_advisors(self):
+ # Test for multiple advisors.
+ grabber = Grabber(19964)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Rudolf Otto Sigismund Lipschitz")
+ self.assertEquals(institution, u"Universit\xe4t Berlin")
+ self.assertEquals(year, 1853)
+ self.assertEquals(advisors, [17946, 47064])
+
+class TestGeneagrapherMethods(unittest.TestCase):
+ """
+ Unit tests for the geneagrapher.Geneagrapher class.
+ """
+ def setUp(self):
+ self.ggrapher = geneagrapher.Geneagrapher()
+
+ def test001_init(self):
+ # Test constructor.
+ self.assertEquals(isinstance(self.ggrapher.graph, Graph), True)
+ self.assertEquals(self.ggrapher.leaf_ids, [])
+ self.assertEquals(self.ggrapher.get_ancestors, False)
+ self.assertEquals(self.ggrapher.get_descendants, False)
+ self.assertEquals(self.ggrapher.verbose, False)
+ self.assertEquals(self.ggrapher.write_filename, None)
+
+ def test002_parse_empty(self):
+ # Test parse_input() with no arguments.
+ sys.argv = ['geneagrapher']
+ self.assertRaises(SyntaxError, self.ggrapher.parse_input)
+
+ def test003_parse_default(self):
+ # Test parse_input() with no options.
+ sys.argv = ['geneagrapher', '3']
+ self.ggrapher.get_ancestors = False
+ self.ggrapher.get_descendents = True
+ self.ggrapher.write_filename = "filler"
+ self.ggrapher.parse_input()
+ self.assertEquals(self.ggrapher.get_ancestors, False)
+ self.assertEquals(self.ggrapher.get_descendants, False)
+ self.assertEquals(self.ggrapher.verbose, False)
+ self.assertEquals(self.ggrapher.write_filename, None)
+ self.assertEquals(self.ggrapher.leaf_ids, [3])
+
+ def test004_parse_options(self):
+ # Test parse_input() with options.
+ sys.argv = ['geneagrapher', '--with-ancestors', '--with-descendants', '--file=filler', '--verbose', '3', '43']
+ self.ggrapher.parse_input()
+ self.assertEquals(self.ggrapher.get_ancestors, True)
+ self.assertEquals(self.ggrapher.get_descendants, True)
+ self.assertEquals(self.ggrapher.verbose, True)
+ self.assertEquals(self.ggrapher.write_filename, "filler")
+ self.assertEquals(self.ggrapher.leaf_ids, [3, 43])
+
+def runTests():
+ suite = unittest.TestSuite()
+ suite.addTest(unittest.makeSuite(TestRecordMethods))
+ suite.addTest(unittest.makeSuite(TestNodeMethods))
+ suite.addTest(unittest.makeSuite(TestGraphMethods))
+ suite.addTest(unittest.makeSuite(TestGrabberMethods))
+ suite.addTest(unittest.makeSuite(TestGeneagrapherMethods))
+ unittest.TextTestRunner(verbosity=1).run(suite)
+
+if __name__ == '__main__':
+ runTests()
diff --git a/tests/test_grabber_methods.py b/tests/test_grabber_methods.py
new file mode 100644
index 0000000..32d11b5
--- /dev/null
+++ b/tests/test_grabber_methods.py
@@ -0,0 +1,455 @@
+import sys
+import unittest
+from geneagrapher import geneagrapher
+from geneagrapher.grabber import Grabber
+from geneagrapher.graph import DuplicateNodeError, Graph, Node, Record
+
+# Unit tests for graph-related classes.
+class TestRecordMethods(unittest.TestCase):
+ """
+ Unit tests for the Record class.
+ """
+ def test001_init(self):
+ # Test the constructor.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assertEqual(record.name, "Carl Friedrich Gauss")
+ self.assertEqual(record.institution, "Universitaet Helmstedt")
+ self.assertEqual(record.year, 1799)
+ self.assertEqual(record.id, 18231)
+
+ def test002_init_bad_name(self):
+ # Test constructor with bad 'name' parameter.
+ self.assertRaises(TypeError, Record, 1, "Universitaet Helmstedt", 1799, 18231)
+
+ def test003_init_bad_institution(self):
+ # Test constructor with bad 'institution' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss", 1, 1799, 18231)
+
+ def test004_init_bad_year(self):
+ # Test constructor with bad 'year' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss",
+ "Universitaet Helmstedt", "1799", 18231)
+
+ def test005_init_bad_id(self):
+ # Test constructor with bad 'id' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss",
+ "Universitaet Helmstedt", 1799, "18231")
+
+ def test006_cmp_equal(self):
+ # Verify two 'equal' records are compared correctly.
+ record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ record2 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record1 == record2)
+
+ def test007_cmp_unequal(self):
+ # Verify two 'unequal' records are compared correctly.
+ record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ record2 = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ self.assert_(record1 < record2)
+
+ def test008_has_institution_yes(self):
+ # Verify has_institution() method returns True when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record.has_institution())
+
+ def test009_has_institution_no(self):
+ # Verify has_institution() method returns False when the conditions are right.
+ record = Record("Carl Friedrich Gauss", None, 1799, 18231)
+ self.assert_(not record.has_institution())
+
+ def test010_has_year_yes(self):
+ # Verify has_year() method returns True when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record.has_year())
+
+ def test011_has_year_no(self):
+ # Verify has_year() method returns False when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", None, 18231)
+ self.assert_(not record.has_year())
+
+class TestNodeMethods(unittest.TestCase):
+ """
+ Unit tests for the Node class.
+ """
+ def setUp(self):
+ self.record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+
+ def test001_init(self):
+ # Test the constructor.
+ node = Node(self.record, [], [])
+ self.assertEquals(node.record, self.record)
+ self.assertEquals(node.ancestors, [])
+ self.assertEquals(node.descendants, [])
+
+ def test002_init_bad_record(self):
+ # Test the constructor for a case where the record passed is not a Record
+ # object.
+ self.assertRaises(TypeError, Node, 1, [], [])
+
+ def test003_init_bad_ancestor_list(self):
+ # Test the constructor for a case where the ancestor list is not a list.
+ self.assertRaises(TypeError, Node, self.record, 1, [])
+
+ def test003_2_init_bad_descendent_list(self):
+ # Test the constructor for a case where the descendent list is not a list.
+ self.assertRaises(TypeError, Node, self.record, [], 1)
+
+ def test004_str_full(self):
+ # Test __str__() method for Node with complete record.
+ node = Node(self.record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test005_str_no_year(self):
+ # Test __str__() method for Node containing record without year.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", None, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\nUniversitaet Helmstedt"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test006_str_no_inst(self):
+ # Test __str__() method for Node containing record without institution.
+ record = Record("Carl Friedrich Gauss", None, 1799, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\n(1799)"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test007_str_no_inst_no_id(self):
+ # Test __str__() method for Node containing record without institution
+ # or year.
+ record = Record("Carl Friedrich Gauss", None, None, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test008_cmp_equal(self):
+ # Test comparison method for Nodes with identical records.
+ record2 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ node1 = Node(self.record, [], [])
+ node2 = Node(record2, [], [])
+ self.assert_(node1 == node2)
+
+ def test009_cmp_unequal(self):
+ # Test comparison method for Nodes with different records.
+ record2 = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ node1 = Node(self.record, [], [])
+ node2 = Node(record2, [], [])
+ self.assert_(node1 < node2)
+
+ def test010_add_ancestor(self):
+ # Test the add_ancestor() method.
+ node = Node(self.record, [], [])
+ node.add_ancestor(5)
+ self.assertEquals(node.ancestors, [5])
+
+ def test011_add_ancestor_bad_type(self):
+ # Test the add_ancestor() method for a case where the parameter type is incorrect.
+ node = Node(self.record, [], [])
+ self.assertRaises(TypeError, node.add_ancestor, '5')
+
+ def test012_get_id(self):
+ node = Node(self.record, [], [])
+ self.assertEquals(node.get_id(), 18231)
+
+ def test013_set_id(self):
+ # Test the set_id() method.
+ node = Node(self.record, [], [])
+ self.assertEquals(node.get_id(), 18231)
+ node.set_id(15)
+ self.assertEquals(node.get_id(), 15)
+
+
+class TestGraphMethods(unittest.TestCase):
+ """
+ Unit tests for the Graph class.
+ """
+ def setUp(self):
+ self.record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.node1 = Node(self.record1, [], [])
+ self.graph1 = Graph([self.node1])
+
+ def test001_init_empty(self):
+ # Test the constructor.
+ graph = Graph()
+ self.assertEquals(graph.heads, None)
+
+ def test002_init(self):
+ # Test the constructor.
+ self.assert_(self.graph1.heads == [self.node1])
+ self.assertEquals(self.graph1.nodes.keys(), [18231])
+ self.assertEquals(self.graph1.nodes[18231], self.node1)
+
+ def test003_init_bad_heads(self):
+ # Test the constructor when passed a bad type for the heads parameter.
+ self.assertRaises(TypeError, Graph, 3)
+
+ def test004_has_node_true(self):
+ # Test the has_node() method for a True case.
+ self.assertEquals(self.graph1.has_node(18231), True)
+
+ def test005_has_node_false(self):
+ # Test the has_node() method for a False case.
+ self.assertEquals(self.graph1.has_node(1), False)
+
+ def test006_get_node(self):
+ # Test the get_node() method.
+ node = self.graph1.get_node(18231)
+ self.assert_(node == self.node1)
+
+ def test007_get_node_not_found(self):
+ # Test the get_node() method for a case where the node does not exist.
+ node = self.graph1.get_node(1)
+ self.assertEquals(node, None)
+
+ def test008_get_node_list(self):
+ # Test the get_node_list() method.
+ self.assertEquals(self.graph1.get_node_list(), [18231])
+
+ def test008_get_node_list_empty(self):
+ # Test the get_node_list() method for an empty graph.
+ graph = Graph()
+ self.assertEquals(graph.get_node_list(), [])
+
+ def test009_add_node(self):
+ # Test the add_node() method.
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1])
+
+ def test010_add_second_node_head(self):
+ # Test the add_node() method when adding a second node and
+ # marking it as a head node.
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [], True)
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1, self.graph1.get_node(38586)])
+
+ def test011_add_node_head(self):
+ # Test the add_node() method when no heads exist.
+ graph = Graph()
+ self.assertEquals(graph.heads, None)
+ graph.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals(graph.heads, [graph.get_node(38586)])
+
+ def test012_add_node_already_present(self):
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertRaises(DuplicateNodeError, self.graph1.add_node, "Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+
+ def test013_add_node_object(self):
+ # Test the add_node_object() method.
+ record = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ node = Node(record, [], [])
+ self.graph1.add_node_object(node)
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1])
+
+ def test014_generate_dot_file(self):
+ # Test the generate_dot_file() method.
+ dotfileexpt = """digraph genealogy {
+ graph [charset="utf-8"];
+ node [shape=plaintext];
+ edge [style=bold];
+
+ 18231 [label="Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"];
+
+}
+"""
+ dotfile = self.graph1.generate_dot_file(True, False)
+ self.assertEquals(dotfile, dotfileexpt)
+
+ def test015_generate_dot_file(self):
+ # Test the generate_dot_file() method.
+ graph = Graph()
+ graph.add_node("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231, [18230], [])
+ graph.add_node("Johann Friedrich Pfaff", "Georg-August-Universitaet Goettingen", 1786, 18230, [66476], [])
+ graph.add_node("Abraham Gotthelf Kaestner", "Universitaet Leipzig", 1739, 66476, [57670], [])
+ graph.add_node("Christian August Hausen", "Martin-Luther-Universitaet Halle-Wittenberg", 1713, 57670, [72669], [])
+ graph.add_node("Johann Christoph Wichmannshausen", "Universitaet Leipzig", 1685, 72669, [21235], [])
+ graph.add_node("Otto Mencke", "Universitaet Leipzig", 1665, 21235, [], [])
+
+ dotfileexpt = """digraph genealogy {
+ graph [charset="utf-8"];
+ node [shape=plaintext];
+ edge [style=bold];
+
+ 18231 [label="Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"];
+ 18230 [label="Johann Friedrich Pfaff \\nGeorg-August-Universitaet Goettingen (1786)"];
+ 66476 [label="Abraham Gotthelf Kaestner \\nUniversitaet Leipzig (1739)"];
+ 57670 [label="Christian August Hausen \\nMartin-Luther-Universitaet Halle-Wittenberg (1713)"];
+ 72669 [label="Johann Christoph Wichmannshausen \\nUniversitaet Leipzig (1685)"];
+ 21235 [label="Otto Mencke \\nUniversitaet Leipzig (1665)"];
+
+ 18230 -> 18231;
+ 66476 -> 18230;
+ 57670 -> 66476;
+ 72669 -> 57670;
+ 21235 -> 72669;
+}
+"""
+ dotfile = graph.generate_dot_file(True, False)
+ self.assertEquals(dotfile, dotfileexpt)
+
+class TestGrabberMethods(unittest.TestCase):
+ """
+ Unit tests for the Grabber class.
+ """
+ def setUp(self):
+ self.grabber = Grabber(18231)
+
+ def test001_init(self):
+ # Test constructor.
+ self.assertEquals(self.grabber.id, 18231)
+ self.assertEquals(self.grabber.pagestr, None)
+ self.assertEquals(self.grabber.name, None)
+ self.assertEquals(self.grabber.institution, None)
+ self.assertEquals(self.grabber.year, None)
+ self.assertEquals(self.grabber.advisors, [])
+ self.assertEquals(self.grabber.descendants, [])
+
+ def test002_get_page(self):
+ # Test get_page() method.
+ self.grabber.get_page()
+ self.assert_(self.grabber.pagestr is not None)
+ self.assert_(u"<title>The Mathematics Genealogy Project - Carl Gau\xdf</title>" in self.grabber.pagestr)
+ # Get page again and test for adverse affects.
+ self.grabber.get_page()
+ self.assert_(u"<title>The Mathematics Genealogy Project - Carl Gau\xdf</title>" in self.grabber.pagestr)
+
+ def test003_extract_info_bad(self):
+ # Verify exception thrown for bad id.
+ grabber = Grabber(999999999)
+ self.assertRaises(ValueError, grabber.extract_node_information)
+
+ def test004_extract_info_all_fields(self):
+ # Test the extract_node_information() method for a record containing all fields.
+ [name, institution, year, advisors, descendents] = self.grabber.extract_node_information()
+ self.assertEquals(name, self.grabber.name)
+ self.assertEquals(institution, self.grabber.institution)
+ self.assertEquals(year, self.grabber.year)
+ self.assertEquals(advisors, self.grabber.advisors)
+ self.assertEquals(name, u"Carl Friedrich Gau\xdf")
+ self.assertEquals(institution, u"Universit\xe4t Helmstedt")
+ self.assertEquals(year, 1799)
+ self.assertEquals(advisors, [18230])
+ self.assertEquals(descendents, [18603, 18233, 62547, 29642, 55175, 29458, 19953, 18232, 151876])
+
+ # Verify calling extract_node_information() twice does not have side effect.
+ [name, institution, year, advisors, descendents] = self.grabber.extract_node_information()
+ self.assertEquals(name, u"Carl Friedrich Gau\xdf")
+ self.assertEquals(institution, u"Universit\xe4t Helmstedt")
+ self.assertEquals(year, 1799)
+ self.assertEquals(advisors, [18230])
+ self.assertEquals(descendents, [18603, 18233, 62547, 29642, 55175, 29458, 19953, 18232, 151876])
+
+ def test005_extract_info_no_advisor(self):
+ # Test the extract_node_information() method for a record with no advisor.
+ grabber = Grabber(137717)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Valentin Alberti")
+ self.assertEquals(institution, u"Universit\xe4t Leipzig")
+ self.assertEquals(year, 1678)
+ self.assertEquals(advisors, [])
+ self.assertEquals(descendents, [127946])
+
+ def test006_extract_info_no_year(self):
+ # Test the extract_node_information() method for a record with no year.
+ # This example also has no descendents.
+ grabber = Grabber(53658)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"S. Cingolani")
+ self.assertEquals(institution, u"Scuola Normale Superiore di Pisa")
+ self.assertEquals(year, None)
+ self.assertEquals(advisors, [51261])
+ self.assertEquals(descendents, [])
+
+ def test007_extract_info_no_inst(self):
+ # Test the extract_node_information() method for a record with no institution.
+ # This test is also missing additional information already tested.
+ grabber = Grabber(52965)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Walter Mayer")
+ self.assertEquals(institution, None)
+ self.assertEquals(year, None)
+ self.assertEquals(advisors, [])
+ self.assertEquals(descendents, [52996])
+
+ # Tests for special (from my point of view) characters:
+ def test008_slash_l(self):
+ # Test the extract_node_information() method for a record
+ # containing a slash l character. Example:
+ # http://www.genealogy.math.ndsu.nodak.edu/id.php?id=7383.
+ grabber = Grabber(7383)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"W\u0142adys\u0142aw Hugo Dyonizy Steinhaus")
+ self.assertEquals(institution, u"Georg-August-Universit\xe4t G\xf6ttingen")
+ self.assertEquals(year, 1911)
+ self.assertEquals(advisors, [7298])
+
+ def test009_multiple_advisors(self):
+ # Test for multiple advisors.
+ grabber = Grabber(19964)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Rudolf Otto Sigismund Lipschitz")
+ self.assertEquals(institution, u"Universit\xe4t Berlin")
+ self.assertEquals(year, 1853)
+ self.assertEquals(advisors, [17946, 47064])
+
+class TestGeneagrapherMethods(unittest.TestCase):
+ """
+ Unit tests for the geneagrapher.Geneagrapher class.
+ """
+ def setUp(self):
+ self.ggrapher = geneagrapher.Geneagrapher()
+
+ def test001_init(self):
+ # Test constructor.
+ self.assertEquals(isinstance(self.ggrapher.graph, Graph), True)
+ self.assertEquals(self.ggrapher.leaf_ids, [])
+ self.assertEquals(self.ggrapher.get_ancestors, False)
+ self.assertEquals(self.ggrapher.get_descendants, False)
+ self.assertEquals(self.ggrapher.verbose, False)
+ self.assertEquals(self.ggrapher.write_filename, None)
+
+ def test002_parse_empty(self):
+ # Test parse_input() with no arguments.
+ sys.argv = ['geneagrapher']
+ self.assertRaises(SyntaxError, self.ggrapher.parse_input)
+
+ def test003_parse_default(self):
+ # Test parse_input() with no options.
+ sys.argv = ['geneagrapher', '3']
+ self.ggrapher.get_ancestors = False
+ self.ggrapher.get_descendents = True
+ self.ggrapher.write_filename = "filler"
+ self.ggrapher.parse_input()
+ self.assertEquals(self.ggrapher.get_ancestors, False)
+ self.assertEquals(self.ggrapher.get_descendants, False)
+ self.assertEquals(self.ggrapher.verbose, False)
+ self.assertEquals(self.ggrapher.write_filename, None)
+ self.assertEquals(self.ggrapher.leaf_ids, [3])
+
+ def test004_parse_options(self):
+ # Test parse_input() with options.
+ sys.argv = ['geneagrapher', '--with-ancestors', '--with-descendants', '--file=filler', '--verbose', '3', '43']
+ self.ggrapher.parse_input()
+ self.assertEquals(self.ggrapher.get_ancestors, True)
+ self.assertEquals(self.ggrapher.get_descendants, True)
+ self.assertEquals(self.ggrapher.verbose, True)
+ self.assertEquals(self.ggrapher.write_filename, "filler")
+ self.assertEquals(self.ggrapher.leaf_ids, [3, 43])
+
+def runTests():
+ suite = unittest.TestSuite()
+ suite.addTest(unittest.makeSuite(TestRecordMethods))
+ suite.addTest(unittest.makeSuite(TestNodeMethods))
+ suite.addTest(unittest.makeSuite(TestGraphMethods))
+ suite.addTest(unittest.makeSuite(TestGrabberMethods))
+ suite.addTest(unittest.makeSuite(TestGeneagrapherMethods))
+ unittest.TextTestRunner(verbosity=1).run(suite)
+
+if __name__ == '__main__':
+ runTests()
diff --git a/tests/test_graph_methods.py b/tests/test_graph_methods.py
new file mode 100644
index 0000000..32d11b5
--- /dev/null
+++ b/tests/test_graph_methods.py
@@ -0,0 +1,455 @@
+import sys
+import unittest
+from geneagrapher import geneagrapher
+from geneagrapher.grabber import Grabber
+from geneagrapher.graph import DuplicateNodeError, Graph, Node, Record
+
+# Unit tests for graph-related classes.
+class TestRecordMethods(unittest.TestCase):
+ """
+ Unit tests for the Record class.
+ """
+ def test001_init(self):
+ # Test the constructor.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assertEqual(record.name, "Carl Friedrich Gauss")
+ self.assertEqual(record.institution, "Universitaet Helmstedt")
+ self.assertEqual(record.year, 1799)
+ self.assertEqual(record.id, 18231)
+
+ def test002_init_bad_name(self):
+ # Test constructor with bad 'name' parameter.
+ self.assertRaises(TypeError, Record, 1, "Universitaet Helmstedt", 1799, 18231)
+
+ def test003_init_bad_institution(self):
+ # Test constructor with bad 'institution' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss", 1, 1799, 18231)
+
+ def test004_init_bad_year(self):
+ # Test constructor with bad 'year' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss",
+ "Universitaet Helmstedt", "1799", 18231)
+
+ def test005_init_bad_id(self):
+ # Test constructor with bad 'id' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss",
+ "Universitaet Helmstedt", 1799, "18231")
+
+ def test006_cmp_equal(self):
+ # Verify two 'equal' records are compared correctly.
+ record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ record2 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record1 == record2)
+
+ def test007_cmp_unequal(self):
+ # Verify two 'unequal' records are compared correctly.
+ record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ record2 = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ self.assert_(record1 < record2)
+
+ def test008_has_institution_yes(self):
+ # Verify has_institution() method returns True when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record.has_institution())
+
+ def test009_has_institution_no(self):
+ # Verify has_institution() method returns False when the conditions are right.
+ record = Record("Carl Friedrich Gauss", None, 1799, 18231)
+ self.assert_(not record.has_institution())
+
+ def test010_has_year_yes(self):
+ # Verify has_year() method returns True when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record.has_year())
+
+ def test011_has_year_no(self):
+ # Verify has_year() method returns False when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", None, 18231)
+ self.assert_(not record.has_year())
+
+class TestNodeMethods(unittest.TestCase):
+ """
+ Unit tests for the Node class.
+ """
+ def setUp(self):
+ self.record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+
+ def test001_init(self):
+ # Test the constructor.
+ node = Node(self.record, [], [])
+ self.assertEquals(node.record, self.record)
+ self.assertEquals(node.ancestors, [])
+ self.assertEquals(node.descendants, [])
+
+ def test002_init_bad_record(self):
+ # Test the constructor for a case where the record passed is not a Record
+ # object.
+ self.assertRaises(TypeError, Node, 1, [], [])
+
+ def test003_init_bad_ancestor_list(self):
+ # Test the constructor for a case where the ancestor list is not a list.
+ self.assertRaises(TypeError, Node, self.record, 1, [])
+
+ def test003_2_init_bad_descendent_list(self):
+ # Test the constructor for a case where the descendent list is not a list.
+ self.assertRaises(TypeError, Node, self.record, [], 1)
+
+ def test004_str_full(self):
+ # Test __str__() method for Node with complete record.
+ node = Node(self.record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test005_str_no_year(self):
+ # Test __str__() method for Node containing record without year.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", None, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\nUniversitaet Helmstedt"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test006_str_no_inst(self):
+ # Test __str__() method for Node containing record without institution.
+ record = Record("Carl Friedrich Gauss", None, 1799, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\n(1799)"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test007_str_no_inst_no_id(self):
+ # Test __str__() method for Node containing record without institution
+ # or year.
+ record = Record("Carl Friedrich Gauss", None, None, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test008_cmp_equal(self):
+ # Test comparison method for Nodes with identical records.
+ record2 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ node1 = Node(self.record, [], [])
+ node2 = Node(record2, [], [])
+ self.assert_(node1 == node2)
+
+ def test009_cmp_unequal(self):
+ # Test comparison method for Nodes with different records.
+ record2 = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ node1 = Node(self.record, [], [])
+ node2 = Node(record2, [], [])
+ self.assert_(node1 < node2)
+
+ def test010_add_ancestor(self):
+ # Test the add_ancestor() method.
+ node = Node(self.record, [], [])
+ node.add_ancestor(5)
+ self.assertEquals(node.ancestors, [5])
+
+ def test011_add_ancestor_bad_type(self):
+ # Test the add_ancestor() method for a case where the parameter type is incorrect.
+ node = Node(self.record, [], [])
+ self.assertRaises(TypeError, node.add_ancestor, '5')
+
+ def test012_get_id(self):
+ node = Node(self.record, [], [])
+ self.assertEquals(node.get_id(), 18231)
+
+ def test013_set_id(self):
+ # Test the set_id() method.
+ node = Node(self.record, [], [])
+ self.assertEquals(node.get_id(), 18231)
+ node.set_id(15)
+ self.assertEquals(node.get_id(), 15)
+
+
+class TestGraphMethods(unittest.TestCase):
+ """
+ Unit tests for the Graph class.
+ """
+ def setUp(self):
+ self.record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.node1 = Node(self.record1, [], [])
+ self.graph1 = Graph([self.node1])
+
+ def test001_init_empty(self):
+ # Test the constructor.
+ graph = Graph()
+ self.assertEquals(graph.heads, None)
+
+ def test002_init(self):
+ # Test the constructor.
+ self.assert_(self.graph1.heads == [self.node1])
+ self.assertEquals(self.graph1.nodes.keys(), [18231])
+ self.assertEquals(self.graph1.nodes[18231], self.node1)
+
+ def test003_init_bad_heads(self):
+ # Test the constructor when passed a bad type for the heads parameter.
+ self.assertRaises(TypeError, Graph, 3)
+
+ def test004_has_node_true(self):
+ # Test the has_node() method for a True case.
+ self.assertEquals(self.graph1.has_node(18231), True)
+
+ def test005_has_node_false(self):
+ # Test the has_node() method for a False case.
+ self.assertEquals(self.graph1.has_node(1), False)
+
+ def test006_get_node(self):
+ # Test the get_node() method.
+ node = self.graph1.get_node(18231)
+ self.assert_(node == self.node1)
+
+ def test007_get_node_not_found(self):
+ # Test the get_node() method for a case where the node does not exist.
+ node = self.graph1.get_node(1)
+ self.assertEquals(node, None)
+
+ def test008_get_node_list(self):
+ # Test the get_node_list() method.
+ self.assertEquals(self.graph1.get_node_list(), [18231])
+
+ def test008_get_node_list_empty(self):
+ # Test the get_node_list() method for an empty graph.
+ graph = Graph()
+ self.assertEquals(graph.get_node_list(), [])
+
+ def test009_add_node(self):
+ # Test the add_node() method.
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1])
+
+ def test010_add_second_node_head(self):
+ # Test the add_node() method when adding a second node and
+ # marking it as a head node.
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [], True)
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1, self.graph1.get_node(38586)])
+
+ def test011_add_node_head(self):
+ # Test the add_node() method when no heads exist.
+ graph = Graph()
+ self.assertEquals(graph.heads, None)
+ graph.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals(graph.heads, [graph.get_node(38586)])
+
+ def test012_add_node_already_present(self):
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertRaises(DuplicateNodeError, self.graph1.add_node, "Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+
+ def test013_add_node_object(self):
+ # Test the add_node_object() method.
+ record = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ node = Node(record, [], [])
+ self.graph1.add_node_object(node)
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1])
+
+ def test014_generate_dot_file(self):
+ # Test the generate_dot_file() method.
+ dotfileexpt = """digraph genealogy {
+ graph [charset="utf-8"];
+ node [shape=plaintext];
+ edge [style=bold];
+
+ 18231 [label="Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"];
+
+}
+"""
+ dotfile = self.graph1.generate_dot_file(True, False)
+ self.assertEquals(dotfile, dotfileexpt)
+
+ def test015_generate_dot_file(self):
+ # Test the generate_dot_file() method.
+ graph = Graph()
+ graph.add_node("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231, [18230], [])
+ graph.add_node("Johann Friedrich Pfaff", "Georg-August-Universitaet Goettingen", 1786, 18230, [66476], [])
+ graph.add_node("Abraham Gotthelf Kaestner", "Universitaet Leipzig", 1739, 66476, [57670], [])
+ graph.add_node("Christian August Hausen", "Martin-Luther-Universitaet Halle-Wittenberg", 1713, 57670, [72669], [])
+ graph.add_node("Johann Christoph Wichmannshausen", "Universitaet Leipzig", 1685, 72669, [21235], [])
+ graph.add_node("Otto Mencke", "Universitaet Leipzig", 1665, 21235, [], [])
+
+ dotfileexpt = """digraph genealogy {
+ graph [charset="utf-8"];
+ node [shape=plaintext];
+ edge [style=bold];
+
+ 18231 [label="Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"];
+ 18230 [label="Johann Friedrich Pfaff \\nGeorg-August-Universitaet Goettingen (1786)"];
+ 66476 [label="Abraham Gotthelf Kaestner \\nUniversitaet Leipzig (1739)"];
+ 57670 [label="Christian August Hausen \\nMartin-Luther-Universitaet Halle-Wittenberg (1713)"];
+ 72669 [label="Johann Christoph Wichmannshausen \\nUniversitaet Leipzig (1685)"];
+ 21235 [label="Otto Mencke \\nUniversitaet Leipzig (1665)"];
+
+ 18230 -> 18231;
+ 66476 -> 18230;
+ 57670 -> 66476;
+ 72669 -> 57670;
+ 21235 -> 72669;
+}
+"""
+ dotfile = graph.generate_dot_file(True, False)
+ self.assertEquals(dotfile, dotfileexpt)
+
+class TestGrabberMethods(unittest.TestCase):
+ """
+ Unit tests for the Grabber class.
+ """
+ def setUp(self):
+ self.grabber = Grabber(18231)
+
+ def test001_init(self):
+ # Test constructor.
+ self.assertEquals(self.grabber.id, 18231)
+ self.assertEquals(self.grabber.pagestr, None)
+ self.assertEquals(self.grabber.name, None)
+ self.assertEquals(self.grabber.institution, None)
+ self.assertEquals(self.grabber.year, None)
+ self.assertEquals(self.grabber.advisors, [])
+ self.assertEquals(self.grabber.descendants, [])
+
+ def test002_get_page(self):
+ # Test get_page() method.
+ self.grabber.get_page()
+ self.assert_(self.grabber.pagestr is not None)
+ self.assert_(u"<title>The Mathematics Genealogy Project - Carl Gau\xdf</title>" in self.grabber.pagestr)
+ # Get page again and test for adverse affects.
+ self.grabber.get_page()
+ self.assert_(u"<title>The Mathematics Genealogy Project - Carl Gau\xdf</title>" in self.grabber.pagestr)
+
+ def test003_extract_info_bad(self):
+ # Verify exception thrown for bad id.
+ grabber = Grabber(999999999)
+ self.assertRaises(ValueError, grabber.extract_node_information)
+
+ def test004_extract_info_all_fields(self):
+ # Test the extract_node_information() method for a record containing all fields.
+ [name, institution, year, advisors, descendents] = self.grabber.extract_node_information()
+ self.assertEquals(name, self.grabber.name)
+ self.assertEquals(institution, self.grabber.institution)
+ self.assertEquals(year, self.grabber.year)
+ self.assertEquals(advisors, self.grabber.advisors)
+ self.assertEquals(name, u"Carl Friedrich Gau\xdf")
+ self.assertEquals(institution, u"Universit\xe4t Helmstedt")
+ self.assertEquals(year, 1799)
+ self.assertEquals(advisors, [18230])
+ self.assertEquals(descendents, [18603, 18233, 62547, 29642, 55175, 29458, 19953, 18232, 151876])
+
+ # Verify calling extract_node_information() twice does not have side effect.
+ [name, institution, year, advisors, descendents] = self.grabber.extract_node_information()
+ self.assertEquals(name, u"Carl Friedrich Gau\xdf")
+ self.assertEquals(institution, u"Universit\xe4t Helmstedt")
+ self.assertEquals(year, 1799)
+ self.assertEquals(advisors, [18230])
+ self.assertEquals(descendents, [18603, 18233, 62547, 29642, 55175, 29458, 19953, 18232, 151876])
+
+ def test005_extract_info_no_advisor(self):
+ # Test the extract_node_information() method for a record with no advisor.
+ grabber = Grabber(137717)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Valentin Alberti")
+ self.assertEquals(institution, u"Universit\xe4t Leipzig")
+ self.assertEquals(year, 1678)
+ self.assertEquals(advisors, [])
+ self.assertEquals(descendents, [127946])
+
+ def test006_extract_info_no_year(self):
+ # Test the extract_node_information() method for a record with no year.
+ # This example also has no descendents.
+ grabber = Grabber(53658)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"S. Cingolani")
+ self.assertEquals(institution, u"Scuola Normale Superiore di Pisa")
+ self.assertEquals(year, None)
+ self.assertEquals(advisors, [51261])
+ self.assertEquals(descendents, [])
+
+ def test007_extract_info_no_inst(self):
+ # Test the extract_node_information() method for a record with no institution.
+ # This test is also missing additional information already tested.
+ grabber = Grabber(52965)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Walter Mayer")
+ self.assertEquals(institution, None)
+ self.assertEquals(year, None)
+ self.assertEquals(advisors, [])
+ self.assertEquals(descendents, [52996])
+
+ # Tests for special (from my point of view) characters:
+ def test008_slash_l(self):
+ # Test the extract_node_information() method for a record
+ # containing a slash l character. Example:
+ # http://www.genealogy.math.ndsu.nodak.edu/id.php?id=7383.
+ grabber = Grabber(7383)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"W\u0142adys\u0142aw Hugo Dyonizy Steinhaus")
+ self.assertEquals(institution, u"Georg-August-Universit\xe4t G\xf6ttingen")
+ self.assertEquals(year, 1911)
+ self.assertEquals(advisors, [7298])
+
+ def test009_multiple_advisors(self):
+ # Test for multiple advisors.
+ grabber = Grabber(19964)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Rudolf Otto Sigismund Lipschitz")
+ self.assertEquals(institution, u"Universit\xe4t Berlin")
+ self.assertEquals(year, 1853)
+ self.assertEquals(advisors, [17946, 47064])
+
+class TestGeneagrapherMethods(unittest.TestCase):
+ """
+ Unit tests for the geneagrapher.Geneagrapher class.
+ """
+ def setUp(self):
+ self.ggrapher = geneagrapher.Geneagrapher()
+
+ def test001_init(self):
+ # Test constructor.
+ self.assertEquals(isinstance(self.ggrapher.graph, Graph), True)
+ self.assertEquals(self.ggrapher.leaf_ids, [])
+ self.assertEquals(self.ggrapher.get_ancestors, False)
+ self.assertEquals(self.ggrapher.get_descendants, False)
+ self.assertEquals(self.ggrapher.verbose, False)
+ self.assertEquals(self.ggrapher.write_filename, None)
+
+ def test002_parse_empty(self):
+ # Test parse_input() with no arguments.
+ sys.argv = ['geneagrapher']
+ self.assertRaises(SyntaxError, self.ggrapher.parse_input)
+
+ def test003_parse_default(self):
+ # Test parse_input() with no options.
+ sys.argv = ['geneagrapher', '3']
+ self.ggrapher.get_ancestors = False
+ self.ggrapher.get_descendents = True
+ self.ggrapher.write_filename = "filler"
+ self.ggrapher.parse_input()
+ self.assertEquals(self.ggrapher.get_ancestors, False)
+ self.assertEquals(self.ggrapher.get_descendants, False)
+ self.assertEquals(self.ggrapher.verbose, False)
+ self.assertEquals(self.ggrapher.write_filename, None)
+ self.assertEquals(self.ggrapher.leaf_ids, [3])
+
+ def test004_parse_options(self):
+ # Test parse_input() with options.
+ sys.argv = ['geneagrapher', '--with-ancestors', '--with-descendants', '--file=filler', '--verbose', '3', '43']
+ self.ggrapher.parse_input()
+ self.assertEquals(self.ggrapher.get_ancestors, True)
+ self.assertEquals(self.ggrapher.get_descendants, True)
+ self.assertEquals(self.ggrapher.verbose, True)
+ self.assertEquals(self.ggrapher.write_filename, "filler")
+ self.assertEquals(self.ggrapher.leaf_ids, [3, 43])
+
+def runTests():
+ suite = unittest.TestSuite()
+ suite.addTest(unittest.makeSuite(TestRecordMethods))
+ suite.addTest(unittest.makeSuite(TestNodeMethods))
+ suite.addTest(unittest.makeSuite(TestGraphMethods))
+ suite.addTest(unittest.makeSuite(TestGrabberMethods))
+ suite.addTest(unittest.makeSuite(TestGeneagrapherMethods))
+ unittest.TextTestRunner(verbosity=1).run(suite)
+
+if __name__ == '__main__':
+ runTests()
diff --git a/tests/test_node_methods.py b/tests/test_node_methods.py
new file mode 100644
index 0000000..32d11b5
--- /dev/null
+++ b/tests/test_node_methods.py
@@ -0,0 +1,455 @@
+import sys
+import unittest
+from geneagrapher import geneagrapher
+from geneagrapher.grabber import Grabber
+from geneagrapher.graph import DuplicateNodeError, Graph, Node, Record
+
+# Unit tests for graph-related classes.
+class TestRecordMethods(unittest.TestCase):
+ """
+ Unit tests for the Record class.
+ """
+ def test001_init(self):
+ # Test the constructor.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assertEqual(record.name, "Carl Friedrich Gauss")
+ self.assertEqual(record.institution, "Universitaet Helmstedt")
+ self.assertEqual(record.year, 1799)
+ self.assertEqual(record.id, 18231)
+
+ def test002_init_bad_name(self):
+ # Test constructor with bad 'name' parameter.
+ self.assertRaises(TypeError, Record, 1, "Universitaet Helmstedt", 1799, 18231)
+
+ def test003_init_bad_institution(self):
+ # Test constructor with bad 'institution' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss", 1, 1799, 18231)
+
+ def test004_init_bad_year(self):
+ # Test constructor with bad 'year' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss",
+ "Universitaet Helmstedt", "1799", 18231)
+
+ def test005_init_bad_id(self):
+ # Test constructor with bad 'id' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss",
+ "Universitaet Helmstedt", 1799, "18231")
+
+ def test006_cmp_equal(self):
+ # Verify two 'equal' records are compared correctly.
+ record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ record2 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record1 == record2)
+
+ def test007_cmp_unequal(self):
+ # Verify two 'unequal' records are compared correctly.
+ record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ record2 = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ self.assert_(record1 < record2)
+
+ def test008_has_institution_yes(self):
+ # Verify has_institution() method returns True when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record.has_institution())
+
+ def test009_has_institution_no(self):
+ # Verify has_institution() method returns False when the conditions are right.
+ record = Record("Carl Friedrich Gauss", None, 1799, 18231)
+ self.assert_(not record.has_institution())
+
+ def test010_has_year_yes(self):
+ # Verify has_year() method returns True when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record.has_year())
+
+ def test011_has_year_no(self):
+ # Verify has_year() method returns False when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", None, 18231)
+ self.assert_(not record.has_year())
+
+class TestNodeMethods(unittest.TestCase):
+ """
+ Unit tests for the Node class.
+ """
+ def setUp(self):
+ self.record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+
+ def test001_init(self):
+ # Test the constructor.
+ node = Node(self.record, [], [])
+ self.assertEquals(node.record, self.record)
+ self.assertEquals(node.ancestors, [])
+ self.assertEquals(node.descendants, [])
+
+ def test002_init_bad_record(self):
+ # Test the constructor for a case where the record passed is not a Record
+ # object.
+ self.assertRaises(TypeError, Node, 1, [], [])
+
+ def test003_init_bad_ancestor_list(self):
+ # Test the constructor for a case where the ancestor list is not a list.
+ self.assertRaises(TypeError, Node, self.record, 1, [])
+
+ def test003_2_init_bad_descendent_list(self):
+ # Test the constructor for a case where the descendent list is not a list.
+ self.assertRaises(TypeError, Node, self.record, [], 1)
+
+ def test004_str_full(self):
+ # Test __str__() method for Node with complete record.
+ node = Node(self.record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test005_str_no_year(self):
+ # Test __str__() method for Node containing record without year.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", None, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\nUniversitaet Helmstedt"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test006_str_no_inst(self):
+ # Test __str__() method for Node containing record without institution.
+ record = Record("Carl Friedrich Gauss", None, 1799, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\n(1799)"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test007_str_no_inst_no_id(self):
+ # Test __str__() method for Node containing record without institution
+ # or year.
+ record = Record("Carl Friedrich Gauss", None, None, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test008_cmp_equal(self):
+ # Test comparison method for Nodes with identical records.
+ record2 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ node1 = Node(self.record, [], [])
+ node2 = Node(record2, [], [])
+ self.assert_(node1 == node2)
+
+ def test009_cmp_unequal(self):
+ # Test comparison method for Nodes with different records.
+ record2 = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ node1 = Node(self.record, [], [])
+ node2 = Node(record2, [], [])
+ self.assert_(node1 < node2)
+
+ def test010_add_ancestor(self):
+ # Test the add_ancestor() method.
+ node = Node(self.record, [], [])
+ node.add_ancestor(5)
+ self.assertEquals(node.ancestors, [5])
+
+ def test011_add_ancestor_bad_type(self):
+ # Test the add_ancestor() method for a case where the parameter type is incorrect.
+ node = Node(self.record, [], [])
+ self.assertRaises(TypeError, node.add_ancestor, '5')
+
+ def test012_get_id(self):
+ node = Node(self.record, [], [])
+ self.assertEquals(node.get_id(), 18231)
+
+ def test013_set_id(self):
+ # Test the set_id() method.
+ node = Node(self.record, [], [])
+ self.assertEquals(node.get_id(), 18231)
+ node.set_id(15)
+ self.assertEquals(node.get_id(), 15)
+
+
+class TestGraphMethods(unittest.TestCase):
+ """
+ Unit tests for the Graph class.
+ """
+ def setUp(self):
+ self.record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.node1 = Node(self.record1, [], [])
+ self.graph1 = Graph([self.node1])
+
+ def test001_init_empty(self):
+ # Test the constructor.
+ graph = Graph()
+ self.assertEquals(graph.heads, None)
+
+ def test002_init(self):
+ # Test the constructor.
+ self.assert_(self.graph1.heads == [self.node1])
+ self.assertEquals(self.graph1.nodes.keys(), [18231])
+ self.assertEquals(self.graph1.nodes[18231], self.node1)
+
+ def test003_init_bad_heads(self):
+ # Test the constructor when passed a bad type for the heads parameter.
+ self.assertRaises(TypeError, Graph, 3)
+
+ def test004_has_node_true(self):
+ # Test the has_node() method for a True case.
+ self.assertEquals(self.graph1.has_node(18231), True)
+
+ def test005_has_node_false(self):
+ # Test the has_node() method for a False case.
+ self.assertEquals(self.graph1.has_node(1), False)
+
+ def test006_get_node(self):
+ # Test the get_node() method.
+ node = self.graph1.get_node(18231)
+ self.assert_(node == self.node1)
+
+ def test007_get_node_not_found(self):
+ # Test the get_node() method for a case where the node does not exist.
+ node = self.graph1.get_node(1)
+ self.assertEquals(node, None)
+
+ def test008_get_node_list(self):
+ # Test the get_node_list() method.
+ self.assertEquals(self.graph1.get_node_list(), [18231])
+
+ def test008_get_node_list_empty(self):
+ # Test the get_node_list() method for an empty graph.
+ graph = Graph()
+ self.assertEquals(graph.get_node_list(), [])
+
+ def test009_add_node(self):
+ # Test the add_node() method.
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1])
+
+ def test010_add_second_node_head(self):
+ # Test the add_node() method when adding a second node and
+ # marking it as a head node.
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [], True)
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1, self.graph1.get_node(38586)])
+
+ def test011_add_node_head(self):
+ # Test the add_node() method when no heads exist.
+ graph = Graph()
+ self.assertEquals(graph.heads, None)
+ graph.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals(graph.heads, [graph.get_node(38586)])
+
+ def test012_add_node_already_present(self):
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertRaises(DuplicateNodeError, self.graph1.add_node, "Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+
+ def test013_add_node_object(self):
+ # Test the add_node_object() method.
+ record = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ node = Node(record, [], [])
+ self.graph1.add_node_object(node)
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1])
+
+ def test014_generate_dot_file(self):
+ # Test the generate_dot_file() method.
+ dotfileexpt = """digraph genealogy {
+ graph [charset="utf-8"];
+ node [shape=plaintext];
+ edge [style=bold];
+
+ 18231 [label="Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"];
+
+}
+"""
+ dotfile = self.graph1.generate_dot_file(True, False)
+ self.assertEquals(dotfile, dotfileexpt)
+
+ def test015_generate_dot_file(self):
+ # Test the generate_dot_file() method.
+ graph = Graph()
+ graph.add_node("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231, [18230], [])
+ graph.add_node("Johann Friedrich Pfaff", "Georg-August-Universitaet Goettingen", 1786, 18230, [66476], [])
+ graph.add_node("Abraham Gotthelf Kaestner", "Universitaet Leipzig", 1739, 66476, [57670], [])
+ graph.add_node("Christian August Hausen", "Martin-Luther-Universitaet Halle-Wittenberg", 1713, 57670, [72669], [])
+ graph.add_node("Johann Christoph Wichmannshausen", "Universitaet Leipzig", 1685, 72669, [21235], [])
+ graph.add_node("Otto Mencke", "Universitaet Leipzig", 1665, 21235, [], [])
+
+ dotfileexpt = """digraph genealogy {
+ graph [charset="utf-8"];
+ node [shape=plaintext];
+ edge [style=bold];
+
+ 18231 [label="Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"];
+ 18230 [label="Johann Friedrich Pfaff \\nGeorg-August-Universitaet Goettingen (1786)"];
+ 66476 [label="Abraham Gotthelf Kaestner \\nUniversitaet Leipzig (1739)"];
+ 57670 [label="Christian August Hausen \\nMartin-Luther-Universitaet Halle-Wittenberg (1713)"];
+ 72669 [label="Johann Christoph Wichmannshausen \\nUniversitaet Leipzig (1685)"];
+ 21235 [label="Otto Mencke \\nUniversitaet Leipzig (1665)"];
+
+ 18230 -> 18231;
+ 66476 -> 18230;
+ 57670 -> 66476;
+ 72669 -> 57670;
+ 21235 -> 72669;
+}
+"""
+ dotfile = graph.generate_dot_file(True, False)
+ self.assertEquals(dotfile, dotfileexpt)
+
+class TestGrabberMethods(unittest.TestCase):
+ """
+ Unit tests for the Grabber class.
+ """
+ def setUp(self):
+ self.grabber = Grabber(18231)
+
+ def test001_init(self):
+ # Test constructor.
+ self.assertEquals(self.grabber.id, 18231)
+ self.assertEquals(self.grabber.pagestr, None)
+ self.assertEquals(self.grabber.name, None)
+ self.assertEquals(self.grabber.institution, None)
+ self.assertEquals(self.grabber.year, None)
+ self.assertEquals(self.grabber.advisors, [])
+ self.assertEquals(self.grabber.descendants, [])
+
+ def test002_get_page(self):
+ # Test get_page() method.
+ self.grabber.get_page()
+ self.assert_(self.grabber.pagestr is not None)
+ self.assert_(u"<title>The Mathematics Genealogy Project - Carl Gau\xdf</title>" in self.grabber.pagestr)
+ # Get page again and test for adverse affects.
+ self.grabber.get_page()
+ self.assert_(u"<title>The Mathematics Genealogy Project - Carl Gau\xdf</title>" in self.grabber.pagestr)
+
+ def test003_extract_info_bad(self):
+ # Verify exception thrown for bad id.
+ grabber = Grabber(999999999)
+ self.assertRaises(ValueError, grabber.extract_node_information)
+
+ def test004_extract_info_all_fields(self):
+ # Test the extract_node_information() method for a record containing all fields.
+ [name, institution, year, advisors, descendents] = self.grabber.extract_node_information()
+ self.assertEquals(name, self.grabber.name)
+ self.assertEquals(institution, self.grabber.institution)
+ self.assertEquals(year, self.grabber.year)
+ self.assertEquals(advisors, self.grabber.advisors)
+ self.assertEquals(name, u"Carl Friedrich Gau\xdf")
+ self.assertEquals(institution, u"Universit\xe4t Helmstedt")
+ self.assertEquals(year, 1799)
+ self.assertEquals(advisors, [18230])
+ self.assertEquals(descendents, [18603, 18233, 62547, 29642, 55175, 29458, 19953, 18232, 151876])
+
+ # Verify calling extract_node_information() twice does not have side effect.
+ [name, institution, year, advisors, descendents] = self.grabber.extract_node_information()
+ self.assertEquals(name, u"Carl Friedrich Gau\xdf")
+ self.assertEquals(institution, u"Universit\xe4t Helmstedt")
+ self.assertEquals(year, 1799)
+ self.assertEquals(advisors, [18230])
+ self.assertEquals(descendents, [18603, 18233, 62547, 29642, 55175, 29458, 19953, 18232, 151876])
+
+ def test005_extract_info_no_advisor(self):
+ # Test the extract_node_information() method for a record with no advisor.
+ grabber = Grabber(137717)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Valentin Alberti")
+ self.assertEquals(institution, u"Universit\xe4t Leipzig")
+ self.assertEquals(year, 1678)
+ self.assertEquals(advisors, [])
+ self.assertEquals(descendents, [127946])
+
+ def test006_extract_info_no_year(self):
+ # Test the extract_node_information() method for a record with no year.
+ # This example also has no descendents.
+ grabber = Grabber(53658)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"S. Cingolani")
+ self.assertEquals(institution, u"Scuola Normale Superiore di Pisa")
+ self.assertEquals(year, None)
+ self.assertEquals(advisors, [51261])
+ self.assertEquals(descendents, [])
+
+ def test007_extract_info_no_inst(self):
+ # Test the extract_node_information() method for a record with no institution.
+ # This test is also missing additional information already tested.
+ grabber = Grabber(52965)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Walter Mayer")
+ self.assertEquals(institution, None)
+ self.assertEquals(year, None)
+ self.assertEquals(advisors, [])
+ self.assertEquals(descendents, [52996])
+
+ # Tests for special (from my point of view) characters:
+ def test008_slash_l(self):
+ # Test the extract_node_information() method for a record
+ # containing a slash l character. Example:
+ # http://www.genealogy.math.ndsu.nodak.edu/id.php?id=7383.
+ grabber = Grabber(7383)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"W\u0142adys\u0142aw Hugo Dyonizy Steinhaus")
+ self.assertEquals(institution, u"Georg-August-Universit\xe4t G\xf6ttingen")
+ self.assertEquals(year, 1911)
+ self.assertEquals(advisors, [7298])
+
+ def test009_multiple_advisors(self):
+ # Test for multiple advisors.
+ grabber = Grabber(19964)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Rudolf Otto Sigismund Lipschitz")
+ self.assertEquals(institution, u"Universit\xe4t Berlin")
+ self.assertEquals(year, 1853)
+ self.assertEquals(advisors, [17946, 47064])
+
+class TestGeneagrapherMethods(unittest.TestCase):
+ """
+ Unit tests for the geneagrapher.Geneagrapher class.
+ """
+ def setUp(self):
+ self.ggrapher = geneagrapher.Geneagrapher()
+
+ def test001_init(self):
+ # Test constructor.
+ self.assertEquals(isinstance(self.ggrapher.graph, Graph), True)
+ self.assertEquals(self.ggrapher.leaf_ids, [])
+ self.assertEquals(self.ggrapher.get_ancestors, False)
+ self.assertEquals(self.ggrapher.get_descendants, False)
+ self.assertEquals(self.ggrapher.verbose, False)
+ self.assertEquals(self.ggrapher.write_filename, None)
+
+ def test002_parse_empty(self):
+ # Test parse_input() with no arguments.
+ sys.argv = ['geneagrapher']
+ self.assertRaises(SyntaxError, self.ggrapher.parse_input)
+
+ def test003_parse_default(self):
+ # Test parse_input() with no options.
+ sys.argv = ['geneagrapher', '3']
+ self.ggrapher.get_ancestors = False
+ self.ggrapher.get_descendents = True
+ self.ggrapher.write_filename = "filler"
+ self.ggrapher.parse_input()
+ self.assertEquals(self.ggrapher.get_ancestors, False)
+ self.assertEquals(self.ggrapher.get_descendants, False)
+ self.assertEquals(self.ggrapher.verbose, False)
+ self.assertEquals(self.ggrapher.write_filename, None)
+ self.assertEquals(self.ggrapher.leaf_ids, [3])
+
+ def test004_parse_options(self):
+ # Test parse_input() with options.
+ sys.argv = ['geneagrapher', '--with-ancestors', '--with-descendants', '--file=filler', '--verbose', '3', '43']
+ self.ggrapher.parse_input()
+ self.assertEquals(self.ggrapher.get_ancestors, True)
+ self.assertEquals(self.ggrapher.get_descendants, True)
+ self.assertEquals(self.ggrapher.verbose, True)
+ self.assertEquals(self.ggrapher.write_filename, "filler")
+ self.assertEquals(self.ggrapher.leaf_ids, [3, 43])
+
+def runTests():
+ suite = unittest.TestSuite()
+ suite.addTest(unittest.makeSuite(TestRecordMethods))
+ suite.addTest(unittest.makeSuite(TestNodeMethods))
+ suite.addTest(unittest.makeSuite(TestGraphMethods))
+ suite.addTest(unittest.makeSuite(TestGrabberMethods))
+ suite.addTest(unittest.makeSuite(TestGeneagrapherMethods))
+ unittest.TextTestRunner(verbosity=1).run(suite)
+
+if __name__ == '__main__':
+ runTests()
diff --git a/tests/test_record_methods.py b/tests/test_record_methods.py
new file mode 100644
index 0000000..32d11b5
--- /dev/null
+++ b/tests/test_record_methods.py
@@ -0,0 +1,455 @@
+import sys
+import unittest
+from geneagrapher import geneagrapher
+from geneagrapher.grabber import Grabber
+from geneagrapher.graph import DuplicateNodeError, Graph, Node, Record
+
+# Unit tests for graph-related classes.
+class TestRecordMethods(unittest.TestCase):
+ """
+ Unit tests for the Record class.
+ """
+ def test001_init(self):
+ # Test the constructor.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assertEqual(record.name, "Carl Friedrich Gauss")
+ self.assertEqual(record.institution, "Universitaet Helmstedt")
+ self.assertEqual(record.year, 1799)
+ self.assertEqual(record.id, 18231)
+
+ def test002_init_bad_name(self):
+ # Test constructor with bad 'name' parameter.
+ self.assertRaises(TypeError, Record, 1, "Universitaet Helmstedt", 1799, 18231)
+
+ def test003_init_bad_institution(self):
+ # Test constructor with bad 'institution' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss", 1, 1799, 18231)
+
+ def test004_init_bad_year(self):
+ # Test constructor with bad 'year' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss",
+ "Universitaet Helmstedt", "1799", 18231)
+
+ def test005_init_bad_id(self):
+ # Test constructor with bad 'id' parameter.
+ self.assertRaises(TypeError, Record, "Carl Friedrich Gauss",
+ "Universitaet Helmstedt", 1799, "18231")
+
+ def test006_cmp_equal(self):
+ # Verify two 'equal' records are compared correctly.
+ record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ record2 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record1 == record2)
+
+ def test007_cmp_unequal(self):
+ # Verify two 'unequal' records are compared correctly.
+ record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ record2 = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ self.assert_(record1 < record2)
+
+ def test008_has_institution_yes(self):
+ # Verify has_institution() method returns True when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record.has_institution())
+
+ def test009_has_institution_no(self):
+ # Verify has_institution() method returns False when the conditions are right.
+ record = Record("Carl Friedrich Gauss", None, 1799, 18231)
+ self.assert_(not record.has_institution())
+
+ def test010_has_year_yes(self):
+ # Verify has_year() method returns True when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.assert_(record.has_year())
+
+ def test011_has_year_no(self):
+ # Verify has_year() method returns False when the conditions are right.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", None, 18231)
+ self.assert_(not record.has_year())
+
+class TestNodeMethods(unittest.TestCase):
+ """
+ Unit tests for the Node class.
+ """
+ def setUp(self):
+ self.record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+
+ def test001_init(self):
+ # Test the constructor.
+ node = Node(self.record, [], [])
+ self.assertEquals(node.record, self.record)
+ self.assertEquals(node.ancestors, [])
+ self.assertEquals(node.descendants, [])
+
+ def test002_init_bad_record(self):
+ # Test the constructor for a case where the record passed is not a Record
+ # object.
+ self.assertRaises(TypeError, Node, 1, [], [])
+
+ def test003_init_bad_ancestor_list(self):
+ # Test the constructor for a case where the ancestor list is not a list.
+ self.assertRaises(TypeError, Node, self.record, 1, [])
+
+ def test003_2_init_bad_descendent_list(self):
+ # Test the constructor for a case where the descendent list is not a list.
+ self.assertRaises(TypeError, Node, self.record, [], 1)
+
+ def test004_str_full(self):
+ # Test __str__() method for Node with complete record.
+ node = Node(self.record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test005_str_no_year(self):
+ # Test __str__() method for Node containing record without year.
+ record = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", None, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\nUniversitaet Helmstedt"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test006_str_no_inst(self):
+ # Test __str__() method for Node containing record without institution.
+ record = Record("Carl Friedrich Gauss", None, 1799, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss \\n(1799)"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test007_str_no_inst_no_id(self):
+ # Test __str__() method for Node containing record without institution
+ # or year.
+ record = Record("Carl Friedrich Gauss", None, None, 18231)
+ node = Node(record, [], [])
+ nodestr = node.__str__()
+ nodestrexpt = "Carl Friedrich Gauss"
+ self.assertEquals(nodestr, nodestrexpt)
+
+ def test008_cmp_equal(self):
+ # Test comparison method for Nodes with identical records.
+ record2 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ node1 = Node(self.record, [], [])
+ node2 = Node(record2, [], [])
+ self.assert_(node1 == node2)
+
+ def test009_cmp_unequal(self):
+ # Test comparison method for Nodes with different records.
+ record2 = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ node1 = Node(self.record, [], [])
+ node2 = Node(record2, [], [])
+ self.assert_(node1 < node2)
+
+ def test010_add_ancestor(self):
+ # Test the add_ancestor() method.
+ node = Node(self.record, [], [])
+ node.add_ancestor(5)
+ self.assertEquals(node.ancestors, [5])
+
+ def test011_add_ancestor_bad_type(self):
+ # Test the add_ancestor() method for a case where the parameter type is incorrect.
+ node = Node(self.record, [], [])
+ self.assertRaises(TypeError, node.add_ancestor, '5')
+
+ def test012_get_id(self):
+ node = Node(self.record, [], [])
+ self.assertEquals(node.get_id(), 18231)
+
+ def test013_set_id(self):
+ # Test the set_id() method.
+ node = Node(self.record, [], [])
+ self.assertEquals(node.get_id(), 18231)
+ node.set_id(15)
+ self.assertEquals(node.get_id(), 15)
+
+
+class TestGraphMethods(unittest.TestCase):
+ """
+ Unit tests for the Graph class.
+ """
+ def setUp(self):
+ self.record1 = Record("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231)
+ self.node1 = Node(self.record1, [], [])
+ self.graph1 = Graph([self.node1])
+
+ def test001_init_empty(self):
+ # Test the constructor.
+ graph = Graph()
+ self.assertEquals(graph.heads, None)
+
+ def test002_init(self):
+ # Test the constructor.
+ self.assert_(self.graph1.heads == [self.node1])
+ self.assertEquals(self.graph1.nodes.keys(), [18231])
+ self.assertEquals(self.graph1.nodes[18231], self.node1)
+
+ def test003_init_bad_heads(self):
+ # Test the constructor when passed a bad type for the heads parameter.
+ self.assertRaises(TypeError, Graph, 3)
+
+ def test004_has_node_true(self):
+ # Test the has_node() method for a True case.
+ self.assertEquals(self.graph1.has_node(18231), True)
+
+ def test005_has_node_false(self):
+ # Test the has_node() method for a False case.
+ self.assertEquals(self.graph1.has_node(1), False)
+
+ def test006_get_node(self):
+ # Test the get_node() method.
+ node = self.graph1.get_node(18231)
+ self.assert_(node == self.node1)
+
+ def test007_get_node_not_found(self):
+ # Test the get_node() method for a case where the node does not exist.
+ node = self.graph1.get_node(1)
+ self.assertEquals(node, None)
+
+ def test008_get_node_list(self):
+ # Test the get_node_list() method.
+ self.assertEquals(self.graph1.get_node_list(), [18231])
+
+ def test008_get_node_list_empty(self):
+ # Test the get_node_list() method for an empty graph.
+ graph = Graph()
+ self.assertEquals(graph.get_node_list(), [])
+
+ def test009_add_node(self):
+ # Test the add_node() method.
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1])
+
+ def test010_add_second_node_head(self):
+ # Test the add_node() method when adding a second node and
+ # marking it as a head node.
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [], True)
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1, self.graph1.get_node(38586)])
+
+ def test011_add_node_head(self):
+ # Test the add_node() method when no heads exist.
+ graph = Graph()
+ self.assertEquals(graph.heads, None)
+ graph.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals(graph.heads, [graph.get_node(38586)])
+
+ def test012_add_node_already_present(self):
+ self.graph1.add_node("Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertRaises(DuplicateNodeError, self.graph1.add_node, "Leonhard Euler", "Universitaet Basel", 1726, 38586, [], [])
+
+ def test013_add_node_object(self):
+ # Test the add_node_object() method.
+ record = Record("Leonhard Euler", "Universitaet Basel", 1726, 38586)
+ node = Node(record, [], [])
+ self.graph1.add_node_object(node)
+ self.assertEquals([38586, 18231], self.graph1.get_node_list())
+ self.assertEquals(self.graph1.heads, [self.node1])
+
+ def test014_generate_dot_file(self):
+ # Test the generate_dot_file() method.
+ dotfileexpt = """digraph genealogy {
+ graph [charset="utf-8"];
+ node [shape=plaintext];
+ edge [style=bold];
+
+ 18231 [label="Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"];
+
+}
+"""
+ dotfile = self.graph1.generate_dot_file(True, False)
+ self.assertEquals(dotfile, dotfileexpt)
+
+ def test015_generate_dot_file(self):
+ # Test the generate_dot_file() method.
+ graph = Graph()
+ graph.add_node("Carl Friedrich Gauss", "Universitaet Helmstedt", 1799, 18231, [18230], [])
+ graph.add_node("Johann Friedrich Pfaff", "Georg-August-Universitaet Goettingen", 1786, 18230, [66476], [])
+ graph.add_node("Abraham Gotthelf Kaestner", "Universitaet Leipzig", 1739, 66476, [57670], [])
+ graph.add_node("Christian August Hausen", "Martin-Luther-Universitaet Halle-Wittenberg", 1713, 57670, [72669], [])
+ graph.add_node("Johann Christoph Wichmannshausen", "Universitaet Leipzig", 1685, 72669, [21235], [])
+ graph.add_node("Otto Mencke", "Universitaet Leipzig", 1665, 21235, [], [])
+
+ dotfileexpt = """digraph genealogy {
+ graph [charset="utf-8"];
+ node [shape=plaintext];
+ edge [style=bold];
+
+ 18231 [label="Carl Friedrich Gauss \\nUniversitaet Helmstedt (1799)"];
+ 18230 [label="Johann Friedrich Pfaff \\nGeorg-August-Universitaet Goettingen (1786)"];
+ 66476 [label="Abraham Gotthelf Kaestner \\nUniversitaet Leipzig (1739)"];
+ 57670 [label="Christian August Hausen \\nMartin-Luther-Universitaet Halle-Wittenberg (1713)"];
+ 72669 [label="Johann Christoph Wichmannshausen \\nUniversitaet Leipzig (1685)"];
+ 21235 [label="Otto Mencke \\nUniversitaet Leipzig (1665)"];
+
+ 18230 -> 18231;
+ 66476 -> 18230;
+ 57670 -> 66476;
+ 72669 -> 57670;
+ 21235 -> 72669;
+}
+"""
+ dotfile = graph.generate_dot_file(True, False)
+ self.assertEquals(dotfile, dotfileexpt)
+
+class TestGrabberMethods(unittest.TestCase):
+ """
+ Unit tests for the Grabber class.
+ """
+ def setUp(self):
+ self.grabber = Grabber(18231)
+
+ def test001_init(self):
+ # Test constructor.
+ self.assertEquals(self.grabber.id, 18231)
+ self.assertEquals(self.grabber.pagestr, None)
+ self.assertEquals(self.grabber.name, None)
+ self.assertEquals(self.grabber.institution, None)
+ self.assertEquals(self.grabber.year, None)
+ self.assertEquals(self.grabber.advisors, [])
+ self.assertEquals(self.grabber.descendants, [])
+
+ def test002_get_page(self):
+ # Test get_page() method.
+ self.grabber.get_page()
+ self.assert_(self.grabber.pagestr is not None)
+ self.assert_(u"<title>The Mathematics Genealogy Project - Carl Gau\xdf</title>" in self.grabber.pagestr)
+ # Get page again and test for adverse affects.
+ self.grabber.get_page()
+ self.assert_(u"<title>The Mathematics Genealogy Project - Carl Gau\xdf</title>" in self.grabber.pagestr)
+
+ def test003_extract_info_bad(self):
+ # Verify exception thrown for bad id.
+ grabber = Grabber(999999999)
+ self.assertRaises(ValueError, grabber.extract_node_information)
+
+ def test004_extract_info_all_fields(self):
+ # Test the extract_node_information() method for a record containing all fields.
+ [name, institution, year, advisors, descendents] = self.grabber.extract_node_information()
+ self.assertEquals(name, self.grabber.name)
+ self.assertEquals(institution, self.grabber.institution)
+ self.assertEquals(year, self.grabber.year)
+ self.assertEquals(advisors, self.grabber.advisors)
+ self.assertEquals(name, u"Carl Friedrich Gau\xdf")
+ self.assertEquals(institution, u"Universit\xe4t Helmstedt")
+ self.assertEquals(year, 1799)
+ self.assertEquals(advisors, [18230])
+ self.assertEquals(descendents, [18603, 18233, 62547, 29642, 55175, 29458, 19953, 18232, 151876])
+
+ # Verify calling extract_node_information() twice does not have side effect.
+ [name, institution, year, advisors, descendents] = self.grabber.extract_node_information()
+ self.assertEquals(name, u"Carl Friedrich Gau\xdf")
+ self.assertEquals(institution, u"Universit\xe4t Helmstedt")
+ self.assertEquals(year, 1799)
+ self.assertEquals(advisors, [18230])
+ self.assertEquals(descendents, [18603, 18233, 62547, 29642, 55175, 29458, 19953, 18232, 151876])
+
+ def test005_extract_info_no_advisor(self):
+ # Test the extract_node_information() method for a record with no advisor.
+ grabber = Grabber(137717)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Valentin Alberti")
+ self.assertEquals(institution, u"Universit\xe4t Leipzig")
+ self.assertEquals(year, 1678)
+ self.assertEquals(advisors, [])
+ self.assertEquals(descendents, [127946])
+
+ def test006_extract_info_no_year(self):
+ # Test the extract_node_information() method for a record with no year.
+ # This example also has no descendents.
+ grabber = Grabber(53658)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"S. Cingolani")
+ self.assertEquals(institution, u"Scuola Normale Superiore di Pisa")
+ self.assertEquals(year, None)
+ self.assertEquals(advisors, [51261])
+ self.assertEquals(descendents, [])
+
+ def test007_extract_info_no_inst(self):
+ # Test the extract_node_information() method for a record with no institution.
+ # This test is also missing additional information already tested.
+ grabber = Grabber(52965)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Walter Mayer")
+ self.assertEquals(institution, None)
+ self.assertEquals(year, None)
+ self.assertEquals(advisors, [])
+ self.assertEquals(descendents, [52996])
+
+ # Tests for special (from my point of view) characters:
+ def test008_slash_l(self):
+ # Test the extract_node_information() method for a record
+ # containing a slash l character. Example:
+ # http://www.genealogy.math.ndsu.nodak.edu/id.php?id=7383.
+ grabber = Grabber(7383)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"W\u0142adys\u0142aw Hugo Dyonizy Steinhaus")
+ self.assertEquals(institution, u"Georg-August-Universit\xe4t G\xf6ttingen")
+ self.assertEquals(year, 1911)
+ self.assertEquals(advisors, [7298])
+
+ def test009_multiple_advisors(self):
+ # Test for multiple advisors.
+ grabber = Grabber(19964)
+ [name, institution, year, advisors, descendents] = grabber.extract_node_information()
+ self.assertEquals(name, u"Rudolf Otto Sigismund Lipschitz")
+ self.assertEquals(institution, u"Universit\xe4t Berlin")
+ self.assertEquals(year, 1853)
+ self.assertEquals(advisors, [17946, 47064])
+
+class TestGeneagrapherMethods(unittest.TestCase):
+ """
+ Unit tests for the geneagrapher.Geneagrapher class.
+ """
+ def setUp(self):
+ self.ggrapher = geneagrapher.Geneagrapher()
+
+ def test001_init(self):
+ # Test constructor.
+ self.assertEquals(isinstance(self.ggrapher.graph, Graph), True)
+ self.assertEquals(self.ggrapher.leaf_ids, [])
+ self.assertEquals(self.ggrapher.get_ancestors, False)
+ self.assertEquals(self.ggrapher.get_descendants, False)
+ self.assertEquals(self.ggrapher.verbose, False)
+ self.assertEquals(self.ggrapher.write_filename, None)
+
+ def test002_parse_empty(self):
+ # Test parse_input() with no arguments.
+ sys.argv = ['geneagrapher']
+ self.assertRaises(SyntaxError, self.ggrapher.parse_input)
+
+ def test003_parse_default(self):
+ # Test parse_input() with no options.
+ sys.argv = ['geneagrapher', '3']
+ self.ggrapher.get_ancestors = False
+ self.ggrapher.get_descendents = True
+ self.ggrapher.write_filename = "filler"
+ self.ggrapher.parse_input()
+ self.assertEquals(self.ggrapher.get_ancestors, False)
+ self.assertEquals(self.ggrapher.get_descendants, False)
+ self.assertEquals(self.ggrapher.verbose, False)
+ self.assertEquals(self.ggrapher.write_filename, None)
+ self.assertEquals(self.ggrapher.leaf_ids, [3])
+
+ def test004_parse_options(self):
+ # Test parse_input() with options.
+ sys.argv = ['geneagrapher', '--with-ancestors', '--with-descendants', '--file=filler', '--verbose', '3', '43']
+ self.ggrapher.parse_input()
+ self.assertEquals(self.ggrapher.get_ancestors, True)
+ self.assertEquals(self.ggrapher.get_descendants, True)
+ self.assertEquals(self.ggrapher.verbose, True)
+ self.assertEquals(self.ggrapher.write_filename, "filler")
+ self.assertEquals(self.ggrapher.leaf_ids, [3, 43])
+
+def runTests():
+ suite = unittest.TestSuite()
+ suite.addTest(unittest.makeSuite(TestRecordMethods))
+ suite.addTest(unittest.makeSuite(TestNodeMethods))
+ suite.addTest(unittest.makeSuite(TestGraphMethods))
+ suite.addTest(unittest.makeSuite(TestGrabberMethods))
+ suite.addTest(unittest.makeSuite(TestGeneagrapherMethods))
+ unittest.TextTestRunner(verbosity=1).run(suite)
+
+if __name__ == '__main__':
+ runTests()
--
Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-science/packages/geneagrapher.git
More information about the debian-science-commits
mailing list