summaryrefslogtreecommitdiff
path: root/tests/merging_tablesTest.py
blob: ed83135f12520ee38bb5e57ae901cb75e976fb1f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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()