# Import folder where sorting algorithms
import sys
import unittest

# For importing from different folders
# OBS: This is supposed to be done with automated testing, hence relative to folder we want to import from
sys.path.append('Algorithms/math/union_of_two_sets')

# If run from local:
# sys.path.append('../../Algorithms/math/union_of_two_sets')

from union_of_two_sets import union

class test_union(unittest.TestCase):

    def setUp(self):
        # test cases we wish to run
        self.L1 = [1,3,5,7,9,10]
        self.L2 = [2,4,6,11,12]
        self.L1L2_correct = [1,2,3,4,5,6,7,9,10,11,12]

        self.L3 = [1,3,5,10]
        self.L4 = [2,4,6,10]
        self.L3L4_correct = [1,2,3,4,5,6,10]

        self.L5 = [1,3,5,10]
        self.L6 = [1,4,6,11]
        self.L5L6_correct = [1,3,4,5,6,10,11]

        self.L7 = [1,2,3,4,5,6,7]
        self.L8 = [1,2,3,4,5,6,7]
        self.L7L8_correct = [1,2,3,4,5,6,7]

        self.L9 = []
        self.L10 = []
        self.L9L10_correct = []


    def test_union_all(self):
        L1L2_output = union(self.L1, self.L2)
        self.assertEqual(L1L2_output, self.L1L2_correct)

    def test_union_lastequal(self):
        L3L4_output = union(self.L3, self.L4)
        self.assertEqual(L3L4_output, self.L3L4_correct)

    def test_union_firstequal(self):
        L5L6_output = union(self.L5, self.L6)
        self.assertEqual(L5L6_output, self.L5L6_correct)

    def test_union_samelist(self):
        L7L8_output = union(self.L7, self.L8)
        self.assertEqual(L7L8_output, self.L7L8_correct)

    def test_union_both_empty(self):
        L9L10_output = union(self.L9, self.L10)
        self.assertEqual(L9L10_output, self.L9L10_correct)

if __name__ == '__main__':
    print("Running sorting tests:")
    unittest.main()