more tests for signal

neurallayer · Feb 22, 2024 · 55b0f21 · 55b0f21
1 parent 27f1179
commit 55b0f21
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/tests/unit/test_signal.py b/tests/unit/test_signal.py
@@ -8,17 +8,24 @@ def test_signal(self):
         s = BUY
         self.assertEqual(1.0, s.rating)
         self.assertTrue(s.is_buy)
+        self.assertFalse(s.is_sell)
+        self.assertTrue(s.is_entry)
+        self.assertTrue(s.is_exit)
         self.assertTrue(SignalType.ENTRY in s.type)
         self.assertTrue(SignalType.EXIT in s.type)
 
+    def test_signal_equal(self):
         x = SELL
         y = Signal(-1.0, SignalType.BOTH)
         self.assertEqual(x, y)
+        self.assertEqual(BUY, BUY)
+        self.assertNotEqual(BUY, SELL)
 
+    def test_signal_rating(self):
         s = Signal(0.5, SignalType.ENTRY)
         self.assertEqual(0.5, s.rating)
-        self.assertTrue(SignalType.ENTRY in s.type)
-        self.assertFalse(SignalType.EXIT in s.type)
+        self.assertTrue(s.is_entry)
+        self.assertFalse(s.is_exit)
 
 
 if __name__ == "__main__":