Having issue with dumping semaphore

pyAmakCore/tests/memory_leak/main.py

@@ -4,25 +4,18 @@ from pyAmakCore.classes.agent import Agent
 
 
 class SimpleAgent(Agent):
-    """
-    test
-    """
+    pass
 
 
 class SimpleAmas(Amas):
-    """
-    test
-    """
 
     def on_initial_agents_creation(self) -> None:
         for i in range(10):
-            self.add_agent(Agent(self))
+            self.add_agent(SimpleAgent(self))
 
 
 class SimpleEnv(Environment):
-    """
-    test
-    """
+    pass
 
 
 env = SimpleEnv()
@@ -31,8 +24,3 @@ amas = SimpleAmas(env)
 amas.put_token()
 
 amas.start()
-
-
-"""
-There are no visible memory leak in pyAmakCore
-"""
\ No newline at end of file

pyAmakCore/tests/test_pickle/main.py

+import pickle
+from time import sleep
+
+from pyAmakCore.classes.amas import Amas
+from pyAmakCore.classes.environment import Environment
+from pyAmakCore.classes.agent import Agent
+
+
+class SimpleAgent(Agent):
+
+    def __init__(self, amas):
+        self.i = 0
+        super().__init__(amas)
+
+    def on_cycle_begin(self) -> None:
+        self.i += 1
+
+class SimpleAmas(Amas):
+
+    def on_initial_agents_creation(self) -> None:
+        for i in range(10):
+            self.add_agent(SimpleAgent(self))
+
+
+    def on_cycle_begin(self) -> None:
+        if self.get_cycle() == 0:
+            self.save()
+
+    def save(self):
+        with open('filename.pickle', 'wb') as handle:
+            pickle.dump(self.get_agents()[0].get_id(), handle, protocol=pickle.HIGHEST_PROTOCOL)
+        sleep(500)
+
+    @classmethod
+    def load(cls, amas_object):
+        with open('filename.pickle', 'wb') as handle:
+            amas_object = pickle.load(handle)
+
+class SimpleEnv(Environment):
+    pass
+
+
+env = SimpleEnv()
+amas = SimpleAmas(env)
+
+amas.put_token()
+
+amas.start()