1 files changed, 34 insertions, 0 deletions
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()