summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/merging_tablesTest.py34
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()