mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-02-25 10:28:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(repr): Return repr as python object

Browse Source
This commit is contained in:
CaedenPH 2023-07-28 21:48:12 +03:00
parent e7284fe5bd
commit 4bb75740be
1 changed files with 27 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
cellular_automata/wa_tor.py
View File

@ -77,18 +77,18 @@ class Entity:
def __repr__(self) -> str: def __repr__(self) -> str:
""" """
>>> Entity(prey=True, coords=(1, 1)) >>> Entity(prey=True, coords=(1, 1))
<entity_type=prey coords=(1, 1) remaining_reproduction_time=5> Entity(prey=True, coords=(1, 1), remaining_reproduction_time=5)
>>> Entity(prey=False, coords=(2, 1)) >>> Entity(prey=False, coords=(2, 1)) # doctest: +NORMALIZE_WHITESPACE
<entity_type=predator coords=(2, 1) remaining_reproduction_time=20 energy=15> Entity(prey=False, coords=(2, 1),
remaining_reproduction_time=20, energy_value=15)
""" """
repr_ = ( repr_ = (
f"entity_type={'prey' if self.prey is True else 'predator'}" f"Entity(prey={self.prey}, coords={self.coords}, "
f" coords={self.coords}" f"remaining_reproduction_time={self.remaining_reproduction_time}"
f" remaining_reproduction_time={self.remaining_reproduction_time}"
) )
if self.prey is False: if self.energy_value is not None:
repr_ += f" energy={self.energy_value}" repr_ += f", energy_value={self.energy_value}"
return f"<{repr_}>" return f"{repr_})"
class WaTor: class WaTor:
@ -230,8 +230,8 @@ class WaTor:
... [None, Entity(True, (2, 1)), None]]) ... [None, Entity(True, (2, 1)), None]])
>>> wt.get_surrounding_prey( >>> wt.get_surrounding_prey(
... Entity(False, (1, 1))) # doctest: +NORMALIZE_WHITESPACE ... Entity(False, (1, 1))) # doctest: +NORMALIZE_WHITESPACE
[<entity_type=prey coords=(2, 1) remaining_reproduction_time=5>, [Entity(prey=True, coords=(2, 1), remaining_reproduction_time=5),
<entity_type=prey coords=(0, 1) remaining_reproduction_time=5>] Entity(prey=True, coords=(0, 1), remaining_reproduction_time=5)]
>>> wt.set_planet([[Entity(False, (0, 0))]]) >>> wt.set_planet([[Entity(False, (0, 0))]])
>>> wt.get_surrounding_prey(Entity(False, (0, 0))) >>> wt.get_surrounding_prey(Entity(False, (0, 0)))
[] []
@ -240,7 +240,7 @@ class WaTor:
... [None, Entity(False, (1, 1)), Entity(True, (2, 1))], ... [None, Entity(False, (1, 1)), Entity(True, (2, 1))],
... [None, None, None]]) ... [None, None, None]])
>>> wt.get_surrounding_prey(Entity(False, (1, 0))) >>> wt.get_surrounding_prey(Entity(False, (1, 0)))
[<entity_type=prey coords=(0, 0) remaining_reproduction_time=5>] [Entity(prey=True, coords=(0, 0), remaining_reproduction_time=5)]
""" """
coords = entity.coords coords = entity.coords
row, col = coords row, col = coords
@ -302,7 +302,7 @@ class WaTor:
>>> wt.set_planet(planet) >>> wt.set_planet(planet)
>>> wt.move_and_reproduce(Entity(True, coords=(1, 1)), direction_orders=["N"]) >>> wt.move_and_reproduce(Entity(True, coords=(1, 1)), direction_orders=["N"])
>>> wt.planet # doctest: +NORMALIZE_WHITESPACE >>> wt.planet # doctest: +NORMALIZE_WHITESPACE
[[None, <entity_type=prey coords=(0, 1) remaining_reproduction_time=4>, None], [[None, Entity(prey=True, coords=(0, 1), remaining_reproduction_time=4), None],
[None, None, None], [None, None, None],
[None, None, None]] [None, None, None]]
>>> wt.planet[0][0] = Entity(True, coords=(0, 0)) >>> wt.planet[0][0] = Entity(True, coords=(0, 0))
@ -310,8 +310,8 @@ class WaTor:
>>> wt.move_and_reproduce(Entity(True, coords=(0, 1)), >>> wt.move_and_reproduce(Entity(True, coords=(0, 1)),
... direction_orders=["N", "W", "E", "S"]) ... direction_orders=["N", "W", "E", "S"])
>>> wt.planet # doctest: +NORMALIZE_WHITESPACE >>> wt.planet # doctest: +NORMALIZE_WHITESPACE
[[<entity_type=prey coords=(0, 0) remaining_reproduction_time=5>, None, [[Entity(prey=True, coords=(0, 0), remaining_reproduction_time=5), None,
<entity_type=prey coords=(0, 2) remaining_reproduction_time=4>], Entity(prey=True, coords=(0, 2), remaining_reproduction_time=4)],
[None, None, None], [None, None, None],
[None, None, None]] [None, None, None]]
>>> wt.planet[0][1] = wt.planet[0][2] >>> wt.planet[0][1] = wt.planet[0][2]
@ -319,8 +319,8 @@ class WaTor:
>>> wt.move_and_reproduce(Entity(True, coords=(0, 1)), >>> wt.move_and_reproduce(Entity(True, coords=(0, 1)),
... direction_orders=["N", "W", "S", "E"]) ... direction_orders=["N", "W", "S", "E"])
>>> wt.planet # doctest: +NORMALIZE_WHITESPACE >>> wt.planet # doctest: +NORMALIZE_WHITESPACE
[[<entity_type=prey coords=(0, 0) remaining_reproduction_time=5>, None, None], [[Entity(prey=True, coords=(0, 0), remaining_reproduction_time=5), None, None],
[None, <entity_type=prey coords=(1, 1) remaining_reproduction_time=4>, None], [None, Entity(prey=True, coords=(1, 1), remaining_reproduction_time=4), None],
[None, None, None]] [None, None, None]]
>>> wt = WaTor(WIDTH, HEIGHT) >>> wt = WaTor(WIDTH, HEIGHT)
@ -330,8 +330,10 @@ class WaTor:
>>> wt.move_and_reproduce(reproducable_entity, >>> wt.move_and_reproduce(reproducable_entity,
... direction_orders=["N", "W", "S", "E"]) ... direction_orders=["N", "W", "S", "E"])
>>> wt.planet # doctest: +NORMALIZE_WHITESPACE >>> wt.planet # doctest: +NORMALIZE_WHITESPACE
[[<entity_type=predator coords=(0, 0) remaining_reproduction_time=20 energy=15>, [[Entity(prey=False, coords=(0, 0),
<entity_type=predator coords=(0, 1) remaining_reproduction_time=20 energy=15>]] remaining_reproduction_time=20, energy_value=15),
Entity(prey=False, coords=(0, 1), remaining_reproduction_time=20,
energy_value=15)]]
""" """
coords = entity.coords coords = entity.coords
row, col = coords row, col = coords
@ -400,8 +402,8 @@ class WaTor:
>>> wt.perform_prey_actions(reproducable_entity, >>> wt.perform_prey_actions(reproducable_entity,
... direction_orders=["N", "W", "S", "E"]) ... direction_orders=["N", "W", "S", "E"])
>>> wt.planet # doctest: +NORMALIZE_WHITESPACE >>> wt.planet # doctest: +NORMALIZE_WHITESPACE
[[<entity_type=prey coords=(0, 0) remaining_reproduction_time=5>, [[Entity(prey=True, coords=(0, 0), remaining_reproduction_time=5),
<entity_type=prey coords=(0, 1) remaining_reproduction_time=5>]] Entity(prey=True, coords=(0, 1), remaining_reproduction_time=5)]]
""" """
self.move_and_reproduce(entity, direction_orders) self.move_and_reproduce(entity, direction_orders)
@ -431,8 +433,8 @@ class WaTor:
>>> wt.set_planet([[Entity(True, coords=(0, 0)), Entity(False, coords=(0, 1))]]) >>> wt.set_planet([[Entity(True, coords=(0, 0)), Entity(False, coords=(0, 1))]])
>>> wt.perform_predator_actions(Entity(False, coords=(0, 1)), (0, 0), []) >>> wt.perform_predator_actions(Entity(False, coords=(0, 1)), (0, 0), [])
>>> wt.planet # doctest: +NORMALIZE_WHITESPACE >>> wt.planet # doctest: +NORMALIZE_WHITESPACE
[[<entity_type=predator coords=(0, 0) [[Entity(prey=False, coords=(0, 0),
remaining_reproduction_time=20 energy=19>, None]] remaining_reproduction_time=20, energy_value=19), None]]
""" """
assert entity.energy_value is not None # [type checking] assert entity.energy_value is not None # [type checking]
@ -576,9 +578,9 @@ def display_visually(wt: WaTor, iter_number: int, *, colour: bool = True) -> Non
if __name__ == "__main__": if __name__ == "__main__":
# import doctest import doctest
# doctest.testmod() doctest.testmod()
wt = WaTor(WIDTH, HEIGHT) wt = WaTor(WIDTH, HEIGHT)
wt.time_passed = display_visually wt.time_passed = display_visually