import unittest

class Test(unittest.TestCase):
    def test_1_prelomi(self):
        besedilo = "V Notranjem stoji vas, Vrh po imenu. V tej vasici je živel v starih časih Krpan, močan in silen človek. Bil je neki tolik, da ga ni kmalu takega. Dela mu ni bilo mar; ampak nosil je od morja na svoji kobilici angleško sol, kar je bilo pa že tistikrat ostro prepovedano. Pazili so ga mejači, da bi ga kje nehotoma zalezli; poštenega boja ž njim so se bali ravno tako kakor pozneje Štempiharja. Krpan se je pa vedno umikal in gledal, da mu niso mogli do živega."
        self.assertEqual(prelomi(besedilo, 80),
"""V Notranjem stoji vas, Vrh po imenu. V tej vasici je živel v starih časih Krpan,
močan in silen človek. Bil je neki tolik, da ga ni kmalu takega. Dela mu ni bilo
mar; ampak nosil je od morja na svoji kobilici angleško sol, kar je bilo pa že
tistikrat ostro prepovedano. Pazili so ga mejači, da bi ga kje nehotoma zalezli;
poštenega boja ž njim so se bali ravno tako kakor pozneje Štempiharja. Krpan se
je pa vedno umikal in gledal, da mu niso mogli do živega.""")
        self.assertEqual(prelomi("abcčd efghi jklmn oprsš", 11), "abcčd efghi\njklmn oprsš")

    def test_2_vojna(self):
        s = [5, 2, 7, 4, 8, 1, 3, 7]
        t = [2, 6, 5, 4, 9]
        s1, t1 = vojna(s, t)
        self.assertEqual(s1, [1, 3, 7, 5, 2, 7, 5])
        self.assertEqual(t1, [6, 2, 9, 8])

        t1, s1 = vojna(t, s)
        self.assertEqual(s1, [1, 3, 7, 5, 2, 7, 5])
        self.assertEqual(t1, [6, 2, 9, 8])

        s = [5, 2, 7, 4, 8]
        t = [2, 6, 5, 4, 9]
        s1, t1 = vojna(s, t)
        self.assertEqual(s1, [5, 2, 7, 5])
        self.assertEqual(t1, [6, 2, 9, 8])

        s = []
        t = [2, 6, 5, 4, 9]
        s1, t1 = vojna(s, t)
        self.assertEqual(s1, [])
        self.assertEqual(t1, [2, 6, 5, 4, 9])

        t1, s1 = vojna(t, s)
        self.assertEqual(s1, [])
        self.assertEqual(t1, [2, 6, 5, 4, 9])

        s = [1, 2, 3, 4]
        t = [1, 2, 3, 4]
        s1, t1 = vojna(s, t)
        self.assertEqual(s1, [])
        self.assertEqual(t1, [])

    def test_3_relacije(self):
        self.assertEqual(
            relacije([("Ana", "Berta"), ("Cilka", "Ema"), ("Ana", "Fanči"),
                      ("Berta", "Ema"), ("Cilka", "Greta")]),
            {"Ana": {"Berta", "Fanči"}, "Berta": {"Ema"},
             "Cilka": {"Ema", "Greta"}})

        a, b, c, d, e, f, g, h, i, j = "abcdefghij"
        tekme = [(a, b), (b, d), (b, e), (e, d), (b, f), (d, i),
                 (e, h), (a, c), (c, e), (c, g), (a, j)]
        self.assertEqual(
            relacije(tekme),
            {a: {b, c, j}, b: {d, e, f}, d: {i}, e: {d, h}, c: {e, g}})

    def test_4_premaga(self):
        relacije = {"Ana": {"Berta", "Fanči"}, "Berta": {"Ema"},
                    "Cilka": {"Ema", "Greta"}}
        imena = ["Ana", "Berta", "Cilka", "Dani", "Ema", "Fanči", "Greta"]
        premagajo = ["Ana Berta", "Ana Fanči", "Ana Ema", "Berta Ema",
                     "Cilka Ema", "Cilka Greta"]

        for kdo in imena:
            for koga in imena:
                if kdo != koga:
                    if f"{kdo} {koga}" in premagajo:
                        self.assertTrue(premaga(kdo, koga, relacije),
                                        f"Napaka: {kdo} premaga {koga}!")
                    else:
                        self.assertFalse(premaga(kdo, koga, relacije),
                                         f"Napaka: {kdo} ne premaga {koga}!")

        crke = "abcdefghij"
        a, b, c, d, e, f, g, h, i, j = crke
        relacije = {a: {b, c, j}, b: {c, d, e, f}, d: {i}, e: {d, h}, c: {e, g}}

        premagajo = "ab ac ad ae af ag ai ah aj bc bd be " \
                    "bf bg bi bh ce cd ci ch cg di ed ei eh"
        for kdo in crke:
            for koga in crke:
                if kdo != koga:
                    if kdo + koga in premagajo:
                        self.assertTrue(premaga(kdo, koga, relacije),
                                        f"Napaka: {kdo} premaga {koga}!")
                    else:
                        self.assertFalse(premaga(kdo, koga, relacije),
                                         f"Napaka: {kdo} ne premaga {koga}!")

    def test_5_kobilica(self):
        k = Kobilica()
        self.assertEqual(k.uravnotezenost(), 0)

        k.dodaj(0, "goba velika", 3)
        k.dodaj(1, "goba mala", 1)
        self.assertEqual(k.uravnotezenost(), -2)

        k.dodaj(1, "brus", 5)
        self.assertEqual(k.uravnotezenost(), 3)

        k.odvzemi("goba mala")
        self.assertEqual(k.uravnotezenost(), 2)


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