import unittest

ovire1 = [(1, 3, 6), (2, 4, 3), (4, 6, 7),
          (3, 4, 9), (6, 9, 5), (9, 10, 2), (9, 10, 8)]

ovire2 = [(1, 3, 6), (2, 4, 3), (4, 6, 7),
          (3, 4, 9), (9, 10, 2), (9, 10, 8)]

ovire3 = [(1, 3, 6), (2, 4, 3),
          (3, 4, 9), (9, 10, 2), (9, 10, 8)]


class Test(unittest.TestCase):
    def test_01_dolzina_ovire(self):
        self.assertEqual(7, dolzina_ovire((4, 10, 3)))
        self.assertEqual(1, dolzina_ovire((5, 5, 12)))

    def test_02_najdaljsa_ovira(self):
        self.assertEqual((6, 9, 5), najdaljsa_ovira(ovire1))
        self.assertEqual((1, 3, 6), najdaljsa_ovira(ovire2))

    def test_03_dimenzije(self):
        self.assertEqual((10, 9), dimenzije(ovire1))
        self.assertEqual((9, 9), dimenzije(ovire1[:-2]))
        self.assertEqual((6, 9), dimenzije(ovire1[:-3]))
        self.assertEqual((3, 6), dimenzije(ovire1[:1]))

    def test_04_globina(self):
        self.assertEqual(3, globina(ovire1, 3))
        self.assertEqual(5, globina(ovire1, 6))
        self.assertEqual(7, globina(ovire2, 6))
        self.assertEqual(10, globina(ovire3, 6))

    def test_05_naj_stolpec(self):
        self.assertEqual(5, naj_stolpec(ovire1))
        self.assertEqual(7, naj_stolpec(ovire2))
        self.assertEqual(5, naj_stolpec(ovire3))

    def test_03_senca(self):
        self.assertEqual([False] * 10, senca(ovire1))
        self.assertEqual([False, False, False, False, False, False, True, True, False, False], senca(ovire2))
        self.assertEqual([False, False, False, False, True, True, True, True, False, False], senca(ovire3))
        self.assertEqual([False] * 6, senca(ovire2[:-3]))
        self.assertEqual([False] * 3, senca(ovire3[:1]))


if __name__ == "__main__":
    unittest.main()