p1/10_1_sledi_v_snegu/naloga_test.py

303 lines
12 KiB
Python
Raw Permalink Normal View History

2024-01-25 16:08:22 +00:00
from .naloga import *
import unittest
from unittest.mock import patch, call
class Test06(unittest.TestCase):
def test_01_pojdi(self):
ana = Kolesar(["." * 5] * 10)
berta = Kolesar(["." * 20] * 3)
self.assertEqual((0, 0), ana.lokacija())
self.assertEqual(0, ana.razdalja())
ana.pojdi(">")
self.assertEqual((1, 0), ana.lokacija())
self.assertEqual(1, ana.razdalja())
ana.pojdi(">")
self.assertEqual((2, 0), ana.lokacija())
self.assertEqual(2, ana.razdalja())
ana.pojdi("v")
self.assertEqual((2, 1), ana.lokacija())
self.assertEqual(3, ana.razdalja())
ana.pojdi("^")
self.assertEqual((2, 0), ana.lokacija())
self.assertEqual(4, ana.razdalja())
ana.pojdi("^") # umre...
self.assertEqual((2, 0), ana.lokacija())
self.assertEqual(4, ana.razdalja())
ana.pojdi(">") # še vedno mrtva
self.assertEqual((2, 0), ana.lokacija())
self.assertEqual(4, ana.razdalja())
self.assertEqual((0, 0), berta.lokacija())
self.assertEqual(0, berta.razdalja())
berta.pojdi("v")
self.assertEqual((0, 1), berta.lokacija())
self.assertEqual(1, berta.razdalja())
berta.pojdi("v")
self.assertEqual((0, 2), berta.lokacija())
self.assertEqual(2, berta.razdalja())
berta.pojdi(">")
self.assertEqual((1, 2), berta.lokacija())
self.assertEqual(3, berta.razdalja())
berta.pojdi("<")
self.assertEqual((0, 2), berta.lokacija())
self.assertEqual(4, berta.razdalja())
berta.pojdi("v") # umre...
self.assertEqual((0, 2), berta.lokacija())
self.assertEqual(4, berta.razdalja())
berta.pojdi(">") # še vedno mrtva
self.assertEqual((0, 2), berta.lokacija())
self.assertEqual(4, berta.razdalja())
ana = Kolesar(["." * 5] * 10)
for _ in range(5):
ana.pojdi(">")
self.assertEqual((4, 0), ana.lokacija())
self.assertEqual(4, ana.razdalja())
ana.pojdi(">")
self.assertEqual((4, 0), ana.lokacija())
self.assertEqual(4, ana.razdalja())
ana.pojdi("v")
self.assertEqual((4, 0), ana.lokacija())
self.assertEqual(4, ana.razdalja())
def test_02_prevozi_klice_pojdi(self):
ana = Kolesar(["." * 5] * 10)
with patch.object(Kolesar, "pojdi", wraps=ana.pojdi) as p:
ana.prevozi(">>vv<")
self.assertEqual((1, 2), ana.lokacija())
self.assertEqual(5, ana.razdalja())
self.assertEqual(5, p.call_count, "Kolesar.prevozi mora klicati Kolesar.pojdi 5-krat")
p.assert_has_calls([call(">"), call(">"), call("v"), call("v"), call("<")])
def test_02_prevozi(self):
ana = Kolesar(["." * 5] * 10)
ana.prevozi(">>vv<")
self.assertEqual((1, 2), ana.lokacija())
self.assertEqual(5, ana.razdalja())
ana = Kolesar(["." * 2000] * 2000)
ana.prevozi(">>5>")
self.assertEqual((7, 0), ana.lokacija())
self.assertEqual(7, ana.razdalja())
ana.prevozi("v>3v")
self.assertEqual((8, 4), ana.lokacija())
self.assertEqual(12, ana.razdalja())
ana = Kolesar(["." * 2000] * 2000)
ana.prevozi(">>12>")
self.assertEqual((14, 0), ana.lokacija())
self.assertEqual(14, ana.razdalja())
ana = Kolesar(["." * 2000] * 2000)
ana.prevozi(">>1941>")
self.assertEqual((1943, 0), ana.lokacija())
self.assertEqual(1943, ana.razdalja())
ana = Kolesar(["." * 2000] * 2000)
ana.prevozi(">>1941><5<")
self.assertEqual((1937, 0), ana.lokacija())
self.assertEqual(1949, ana.razdalja())
ana = Kolesar(["." * 2000] * 2000)
ana.prevozi(">>1941><5<^")
self.assertEqual((1937, 0), ana.lokacija())
self.assertEqual(1949, ana.razdalja())
ana = Kolesar(["." * 2000] * 2000)
ana.prevozi(">>1941><5<^50>")
self.assertEqual((1937, 0), ana.lokacija())
self.assertEqual(1949, ana.razdalja())
ana = Kolesar(["." * 2000] * 2000)
ana.prevozi("3v3>")
self.assertEqual((3, 3), ana.lokacija())
self.assertEqual(6, ana.razdalja())
ana.prevozi("<")
self.assertEqual((2, 3), ana.lokacija())
self.assertEqual(7, ana.razdalja())
ana.prevozi("5<")
self.assertEqual((0, 3), ana.lokacija())
self.assertEqual(9, ana.razdalja())
ana.prevozi("v")
self.assertEqual((0, 3), ana.lokacija())
self.assertEqual(9, ana.razdalja())
ana.prevozi("5v")
self.assertEqual((0, 3), ana.lokacija())
self.assertEqual(9, ana.razdalja())
ana = Kolesar(["." * 2000] * 2000)
ana.prevozi(">>5>12vv6<vv13^<")
self.assertEqual((0, 2), ana.lokacija())
self.assertEqual(42, ana.razdalja())
ana = Kolesar(["." * 2000] * 2000)
ana.prevozi(">>5>12vv6<vv13^<>>>")
self.assertEqual((3, 2), ana.lokacija())
self.assertEqual(45, ana.razdalja())
class Test07(unittest.TestCase):
def test_00_pojdi_klice_podedovano(self):
ana = Zbiralec(1, 1, dravlje)
with patch.object(Kolesar, "pojdi") as p:
ana.pojdi(">")
self.assertEqual(1, p.call_count, "Zbiralec.pojdi mora klicati Kolesar.pojdi")
p.assert_has_calls([call(">")])
def test_00_premik(self):
self.assertIs(Kolesar.prevozi, Zbiralec.prevozi, "Zbiralec naj ne definira svoje metode premik")
def test_01_znacke(self):
ana = Zbiralec(1, 1, dravlje)
self.assertEqual(set(), ana.znacke())
ana.pojdi(">")
self.assertEqual((2, 1), ana.lokacija())
self.assertEqual({"b"}, ana.znacke())
ana.pojdi(">")
self.assertEqual({"b"}, ana.znacke())
ana.prevozi("v2^<")
self.assertEqual({"b"}, ana.znacke())
ana.prevozi("<2>2v4<") # pobere p, vendar se pelje predaleč in umre ...
self.assertEqual({"b", "r", "p"}, ana.znacke())
ana.prevozi("2v3>") # ... zato tu ne dobi s-ja
self.assertEqual({"b", "r", "p"}, ana.znacke())
def test_02_naj_znacke(self):
ana = Zbiralec(3, 2, dravlje)
self.assertEqual(set(), ana.naj_znacke())
ana.pojdi(">") # x = 4
self.assertEqual((4, 2), ana.lokacija())
self.assertEqual(set(), ana.naj_znacke())
ana.pojdi(">") # x = 5
self.assertEqual({"r"}, ana.naj_znacke())
ana.prevozi("5<") # x = 0
self.assertEqual({"r", "p"}, ana.naj_znacke()) # čez r in p je šla enkrat
ana.prevozi("3>") # x = 3
self.assertEqual({"p"}, ana.naj_znacke()) # čez p je šla dvakrat
ana.prevozi("4>") # x = 7
self.assertEqual({"r", "p"}, ana.naj_znacke()) # čez r in p je šla dvakrat
ana.prevozi("3v")
self.assertEqual({"r", "p", "l"}, ana.znacke()) # imamo še l, vendar le enkrat
self.assertEqual({"r", "p"}, ana.naj_znacke()) # čez r in p je šla dvakrat
ana.prevozi("^^^")
self.assertEqual({"r", "p", "l"}, ana.naj_znacke()) # čez vse je šla trikrat
ana.prevozi("3<") # x = 4
self.assertEqual({"r"}, ana.naj_znacke()) # čez r je šla štirikrat
ana.prevozi("2v3>") # x = 7
self.assertEqual({"l", "r"}, ana.naj_znacke()) # čez l tudi
ana.prevozi("7<")
self.assertEqual({"l", "r"}, ana.naj_znacke())
self.assertEqual({"r", "p", "l", "s"}, ana.znacke()) # imamo še l, vendar le enkrat
ana.prevozi("<") # umre
self.assertEqual({"l", "r"}, ana.naj_znacke())
self.assertEqual({"r", "p", "l", "s"}, ana.znacke()) # imamo še l, vendar le enkrat
ana.prevozi("<2>2<2>2<2>2<2>2<") # če bi bila živa, bi vozila čez s, vendar ... ne
self.assertEqual({"l", "r"}, ana.naj_znacke())
def test_03_trofeje(self):
ana = Zbiralec(0, 0, [".abccbdadbebddefc"])
self.assertEqual([], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("a", 1)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("a", 1), ("b", 1)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("a", 1), ("b", 1), ("c", 1)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("c", 2), ("a", 1), ("b", 1)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("b", 2), ("c", 2), ("a", 1)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("b", 2), ("c", 2), ("a", 1), ("d", 1)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("a", 2), ("b", 2), ("c", 2)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("a", 2), ("b", 2), ("c", 2), ("d", 2)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("b", 3), ("a", 2), ("c", 2), ("d", 2)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("b", 3), ("a", 2), ("c", 2), ("d", 2)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("b", 4), ("a", 2), ("c", 2), ("d", 2)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("b", 4), ("d", 3), ("a", 2), ("c", 2)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("b", 4), ("d", 4), ("a", 2), ("c", 2)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("b", 4), ("d", 4), ("a", 2), ("c", 2), ("e", 2)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("b", 4), ("d", 4), ("a", 2), ("c", 2), ("e", 2)], ana.trofeje())
ana.pojdi(">")
self.assertEqual([("b", 4), ("d", 4), ("c", 3)], ana.trofeje())
class Test08(unittest.TestCase):
def test_00_drsalec_klice_super(self):
zemljevid = ["." * 12] * 7
sledi = Sledi()
ana = Drsalec(1, 1, zemljevid, sledi)
berta = Drsalec(0, 1, zemljevid, sledi)
ana.prevozi("5>")
with patch.object(Zbiralec, "pojdi") as pojdi:
berta.prevozi(">")
self.assertEqual(pojdi.call_count, 4, "Drsalčev pojdi bi moral štirikrat poklicati podedovani pojdi")
def test_01_drsalec(self):
zemljevid = ["." * 12] * 7
zemljevid[3] = "....abcdef.."
sledi = Sledi()
ana = Drsalec(1, 2, zemljevid, sledi)
berta = Drsalec(3, 0, zemljevid, sledi)
cilka = Drsalec(7, 3, zemljevid, sledi)
ana.prevozi("7>2v7<")
self.assertEqual((1, 4), ana.lokacija())
self.assertEqual(16, ana.razdalja())
self.assertEqual([("e", 1)], ana.trofeje())
berta.prevozi("3v")
self.assertEqual((6, 3), berta.lokacija())
self.assertEqual(6, berta.razdalja())
self.assertEqual([("c", 1)], berta.trofeje())
berta.prevozi("<<")
self.assertEqual((4, 3), berta.lokacija())
self.assertEqual(8, berta.razdalja())
self.assertEqual([("a", 1), ("b", 1), ("c", 1)], berta.trofeje())
cilka.prevozi("2^1>")
self.assertEqual((9, 4), cilka.lokacija())
self.assertEqual(9, cilka.razdalja())
self.assertEqual([("e", 3)], cilka.trofeje())
dani = Drsalec(9, 3, zemljevid, sledi)
dani.prevozi("<v")
self.assertEqual((9, 5), dani.lokacija())
self.assertEqual(4, dani.razdalja())
self.assertEqual([("e", 1)], dani.trofeje())
cilka.pojdi("^")
self.assertEqual((9, 4), cilka.lokacija())
self.assertEqual(13, cilka.razdalja())
self.assertEqual([("e", 4), ("f", 1)], cilka.trofeje())
dravlje = """
.r......c..c
r.b.........
.p...r......
............
.s.....l....
.ad......a..
............
..gl......g.""".strip().splitlines()
if __name__ == '__main__':
unittest.main()