From be3113ca3694f886921914ac1ce6965184c6d895 Mon Sep 17 00:00:00 2001
From: shinedday <shinedday@gmail.com>
Date: Fri, 21 May 2021 09:39:32 +0200
Subject: [PATCH] Add seed
---
pyAmakCore/classes/environment.py | 12 ++++---
pyAmakCore/tests/seed/main.py | 54 -------------------------------
2 files changed, 7 insertions(+), 59 deletions(-)
delete mode 100644 pyAmakCore/tests/seed/main.py
diff --git a/pyAmakCore/classes/environment.py b/pyAmakCore/classes/environment.py
index b811398..2b4aa0a 100644
--- a/pyAmakCore/classes/environment.py
+++ b/pyAmakCore/classes/environment.py
@@ -16,14 +16,16 @@ class Environment(Schedulable):
Environment class
"""
- def __init__(self) -> None:
- self.set_seed()
+ def __init__(self, seed_int: int = None) -> None:
+ self.set_seed(seed_int)
super().__init__()
self.on_initialization()
- def set_seed(self):
+ def set_seed(self, number):
"""
This method set the seed for all random in the system, it should be override to set a custom seed
"""
- seed()
-
+ if number is None:
+ seed()
+ return
+ seed(number)
diff --git a/pyAmakCore/tests/seed/main.py b/pyAmakCore/tests/seed/main.py
deleted file mode 100644
index 6a81632..0000000
--- a/pyAmakCore/tests/seed/main.py
+++ /dev/null
@@ -1,54 +0,0 @@
-from random import randint, seed
-from time import sleep
-
-from pyAmakCore.classes.amas import Amas
-from pyAmakCore.classes.environment import Environment
-from pyAmakCore.classes.agent import Agent
-from pyAmakCore.exception.override import ToOverrideWarning
-
-
-class SimpleAgent(Agent):
- """
- test
- """
- def on_act(self) -> None:
- print(randint(0, 100))
-
-
-
-class SimpleAmas(Amas):
- """
- test
- """
-
- def on_initialization(self) -> None:
- ToOverrideWarning.enable_warning(False)
-
- def on_initial_agents_creation(self) -> None:
- for i in range(10):
- self.add_agent(SimpleAgent(self))
-
- def on_cycle_begin(self) -> None:
- pass
- # print(randint(0, 100))
-
- def on_cycle_end(self) -> None:
- if self.get_cycle() == 30:
- sleep(300)
-
-
-class SimpleEnv(Environment):
- """
- test
- """
-
- def set_seed(self):
- seed(30)
-
-
-env = SimpleEnv()
-amas = SimpleAmas(env)
-
-amas.put_token()
-
-amas.start()
--
GitLab