10_1_sledi_v_snegu
This commit is contained in:
Gašper Dobrovoljc
parent
45667c29f9
commit
882d8fdb4c
92
10_1_sledi_v_snegu/naloga.py
Normal file
92
10_1_sledi_v_snegu/naloga.py
Normal file
|
|
@ -0,0 +1,92 @@
|
|||
from re import finditer
|
||||
from collections import defaultdict
|
||||
|
||||
|
||||
class Kolesar:
|
||||
def __init__(self, zemljevid: list[str]):
|
||||
self.zemljevid = zemljevid
|
||||
self.x = 0
|
||||
self.y = 0
|
||||
self.dist = 0
|
||||
self.mrtev = False
|
||||
|
||||
def lokacija(self):
|
||||
return self.x, self.y
|
||||
|
||||
def razdalja(self):
|
||||
return self.dist
|
||||
|
||||
def pojdi(self, smer: str):
|
||||
if self.mrtev:
|
||||
return
|
||||
|
||||
x = self.x
|
||||
y = self.y
|
||||
(x, y) = {
|
||||
"<": (x - 1, y),
|
||||
">": (x + 1, y),
|
||||
"^": (x, y - 1),
|
||||
"v": (x, y + 1),
|
||||
}[smer]
|
||||
|
||||
if not (0 <= x < len(self.zemljevid[0]) and 0 <= y < len(self.zemljevid)):
|
||||
self.mrtev = True
|
||||
return
|
||||
|
||||
self.x = x
|
||||
self.y = y
|
||||
self.dist += 1
|
||||
|
||||
def prevozi(self, pot: str):
|
||||
for match in finditer("([0-9]*)([<>v^])", pot):
|
||||
dist, smer = match.groups()
|
||||
dist = int(dist) if dist != "" else 1
|
||||
for _ in range(dist):
|
||||
self.pojdi(smer)
|
||||
|
||||
|
||||
class Zbiralec(Kolesar):
|
||||
def __init__(self, x: int, y: int, zemljevid: list[str]):
|
||||
super().__init__(zemljevid)
|
||||
self.x = x
|
||||
self.y = y
|
||||
self.pobrane = defaultdict(lambda: 0)
|
||||
|
||||
def znacke(self):
|
||||
return self.pobrane.keys()
|
||||
|
||||
def naj_znacke(self):
|
||||
return {znacka for znacka, count, in self.pobrane.items() if count == max(self.pobrane.values())}
|
||||
|
||||
def pojdi(self, smer: str):
|
||||
super().pojdi(smer)
|
||||
znacka = self.zemljevid[self.y][self.x]
|
||||
if znacka != ".":
|
||||
self.pobrane[znacka] += 1
|
||||
|
||||
def trofeje(self):
|
||||
prev = None
|
||||
return [(znacka, prev := count) for i, (znacka, count) in
|
||||
enumerate(sorted(self.pobrane.items(), key=lambda x: (-x[1], x[0])))
|
||||
if prev == count or i <= 2]
|
||||
|
||||
|
||||
class Sledi(dict[tuple[int, int], str]):
|
||||
pass
|
||||
|
||||
|
||||
class Drsalec(Zbiralec):
|
||||
def __init__(self, x: int, y: int, zemljevid: list[str], sledi: Sledi):
|
||||
super().__init__(x, y, zemljevid)
|
||||
self.sledi = sledi
|
||||
|
||||
def pojdi(self, smer: str):
|
||||
self.sledi[(self.x, self.y)] = smer
|
||||
super().pojdi(smer)
|
||||
|
||||
for _ in range(3):
|
||||
sled = self.sledi.get((self.x, self.y))
|
||||
if sled is None:
|
||||
break
|
||||
|
||||
super().pojdi(sled)
|
||||
302
10_1_sledi_v_snegu/naloga_test.py
Normal file
302
10_1_sledi_v_snegu/naloga_test.py
Normal file
|
|
@ -0,0 +1,302 @@
|
|||
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()
|
||||
Loading…
Reference in New Issue
Block a user