1 files changed, 18 insertions, 0 deletions
diff --git a/tests/TestBQS.py b/tests/TestBQS.py
new file mode 100644
index 0000000..8dc90b2
--- /dev/null
+++ b/tests/TestBQS.py
@@ -0,0 +1,18 @@
+import unittest
+import util.BehaviorQuerySystem as bqs
+from behaviors.ColorChangerBehavior import *
+class TestBQS(unittest.TestCase):
+    def setUp(self):
+        bqs.initBQS()
+        b = ColorChangerBehavior({'Id': 'color','ColorList':[(255,0,0)]})
+        bqs.addBehavior(b)
+        b.addInput({'Location':(5,5)})
+        b.timeStep()
+    def tearDown(self):
+        bqs.initBQS()
+
+    def test_color_predicate(self):
+        validQuery = lambda args:args['Color']==(255,0,0)
+        invalidQuery = lambda args:args['Color']==(254,0,0)
+        assert bqs.query(validQuery) == [{'Color':(255,0,0), 'Location':(5,5)}]
+        assert bqs.query(invalidQuery) == []