From 9b38f4a68773bb08fe923136c9faeade6a9794e9 Mon Sep 17 00:00:00 2001 From: Haidong Ji Date: Mon, 4 Mar 2019 20:51:19 -0600 Subject: Merging tables done. Not too bad since I worked it out in Java. A bit surprised that Python class is used in starter file, but I made it testable and wrote test cases. All is well. --- tests/merging_tablesTest.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 tests/merging_tablesTest.py (limited to 'tests') diff --git a/tests/merging_tablesTest.py b/tests/merging_tablesTest.py new file mode 100644 index 0000000..ed83135 --- /dev/null +++ b/tests/merging_tablesTest.py @@ -0,0 +1,34 @@ +import unittest +from sources import merging_tables + + +class MyTestCase(unittest.TestCase): + def test1(self): + counts = [1, 1, 1, 1, 1] + db = merging_tables.Database(counts) + db.merge(4, 2) + self.assertEqual(2, db.max_row_count) + db.merge(3, 1) + self.assertEqual(2, db.max_row_count) + db.merge(3, 0) + self.assertEqual(3, db.max_row_count) + db.merge(3, 4) + self.assertEqual(5, db.max_row_count) + db.merge(2, 4) + self.assertEqual(5, db.max_row_count) + + def test2(self): + counts = [10, 0, 5, 0, 3, 3] + db = merging_tables.Database(counts) + db.merge(5, 5) + self.assertEqual(10, db.max_row_count) + db.merge(4, 5) + self.assertEqual(10, db.max_row_count) + db.merge(3, 4) + self.assertEqual(10, db.max_row_count) + db.merge(2, 3) + self.assertEqual(11, db.max_row_count) + + +if __name__ == '__main__': + unittest.main() -- cgit v1.2.3