Browse Source

move code

Bastien Sevajol 7 years ago
parent
commit
d491d1662c
1 changed files with 32 additions and 28 deletions
  1. 32 28
      tests/test_simulation.py

+ 32 - 28
tests/test_simulation.py View File

49
     behaviours_classes = [MySimulationBehaviour]
49
     behaviours_classes = [MySimulationBehaviour]
50
 
50
 
51
 
51
 
52
+class MyCycledSubjectBehaviour(MySubjectBehaviour):
53
+    @property
54
+    def cycle_frequency(self):
55
+        return 2
56
+
57
+
58
+class MyCycledSimulationBehaviour(MySimulationBehaviour):
59
+    @property
60
+    def cycle_frequency(self):
61
+        return 2
62
+
63
+
64
+class MyTimedSubjectBehaviour(MySubjectBehaviour):
65
+    @property
66
+    def seconds_frequency(self):
67
+        return 1.0
68
+
69
+    def run(self, data):
70
+        self.last_execution_time = time.time()
71
+        return super().run(data)
72
+
73
+
74
+class MyTimedSimulationBehaviour(MySimulationBehaviour):
75
+    @property
76
+    def seconds_frequency(self):
77
+        return 1.0
78
+
79
+    def run(self, data):
80
+        self.last_execution_time = time.time()
81
+        return super().run(data)
82
+
83
+
52
 class TestBehaviours(BaseTest):
84
 class TestBehaviours(BaseTest):
53
     def test_subject_behaviour_produce_data(
85
     def test_subject_behaviour_produce_data(
54
         self,
86
         self,
105
     ):
137
     ):
106
         shared.reset()
138
         shared.reset()
107
 
139
 
108
-        class MyCycledSubjectBehaviour(MySubjectBehaviour):
109
-            @property
110
-            def cycle_frequency(self):
111
-                return 2
112
-
113
         class MySubject(Subject):
140
         class MySubject(Subject):
114
             behaviours_classes = [MyCycledSubjectBehaviour]
141
             behaviours_classes = [MyCycledSubjectBehaviour]
115
 
142
 
146
     ):
173
     ):
147
         shared.reset()
174
         shared.reset()
148
 
175
 
149
-        class MyTimedSubjectBehaviour(MySubjectBehaviour):
150
-            @property
151
-            def seconds_frequency(self):
152
-                return 1.0
153
-
154
-            def run(self, data):
155
-                self.last_execution_time = time.time()
156
-                return super().run(data)
157
-
158
         class MySubject(Subject):
176
         class MySubject(Subject):
159
             behaviours_classes = [MyTimedSubjectBehaviour]
177
             behaviours_classes = [MyTimedSubjectBehaviour]
160
 
178
 
205
     ):
223
     ):
206
         shared.reset()
224
         shared.reset()
207
 
225
 
208
-        class MyCycledSimulationBehaviour(MySimulationBehaviour):
209
-            @property
210
-            def cycle_frequency(self):
211
-                return 2
212
-
213
         class MyCycledSimulation(Simulation):
226
         class MyCycledSimulation(Simulation):
214
             behaviours_classes = [MyCycledSimulationBehaviour]
227
             behaviours_classes = [MyCycledSimulationBehaviour]
215
 
228
 
240
     ):
253
     ):
241
         shared.reset()
254
         shared.reset()
242
 
255
 
243
-        class MyTimedSimulationBehaviour(MySimulationBehaviour):
244
-            @property
245
-            def seconds_frequency(self):
246
-                return 1.0
247
-
248
-            def run(self, data):
249
-                self.last_execution_time = time.time()
250
-                return super().run(data)
251
-
252
         class MyTimedSimulation(Simulation):
256
         class MyTimedSimulation(Simulation):
253
             behaviours_classes = [MyTimedSimulationBehaviour]
257
             behaviours_classes = [MyTimedSimulationBehaviour]
254
 
258