1 files changed, 40 insertions, 0 deletions
diff --git a/tests/TestBQS.py b/tests/TestBQS.py
new file mode 100644
index 0000000..7316c31
--- /dev/null
+++ b/tests/TestBQS.py
@@ -0,0 +1,40 @@
+import unittest
+import util.BehaviorQuerySystem as bqs
+from behaviors.ColorChangerBehavior import *
+import util.Geo as geo
+class TestBQS(unittest.TestCase):
+    def setUp(self):
+        bqs.initBQS()
+        b = ColorChangerBehavior({'Id': 'color','ColorList':[(255,0,0)]})
+        c = ColorChangerBehavior({'Id': 'color2', 'ColorList':[(0,0,255)]})
+        bqs.addBehavior(b)
+        bqs.addBehavior(c)
+        b.addInput({'Location':(3,4)})
+        c.addInput({'Location':(5,12)})
+        b.timeStep()
+        c.timeStep()
+    def tearDown(self):
+        bqs.initBQS()
+    def test_simple_query(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':(3,4)}]
+        assert bqs.query(invalidQuery) == []
+    def test_dist_query(self):
+        validDist = lambda args:geo.dist(args['Location'], (0,0)) <= 5
+        invalidDist = lambda args:geo.dist(args['Location'], (0,0)) <= 2
+        doubleDist = lambda args:geo.dist(args['Location'], (0,0)) <= 20
+
+        assert bqs.query(validDist) == [{'Color':(255,0,0), 'Location':(3,4)}] 
+        assert bqs.query(invalidDist) == [] 
+        assert bqs.query(doubleDist) == [{'Color':(255,0,0), 'Location':(3,4)}, {'Color':(0,0,255),\
+                                         'Location':(5,12)}] 
+    def test_complex_queries(self):
+
+        validQuery = lambda args:args['Color']==(255,0,0)
+        doubleDist = lambda args:geo.dist(args['Location'], (0,0)) <= 20
+        
+        twoPartPredicate = lambda args:doubleDist(args) and validQuery(args)
+        assert bqs.query(twoPartPredicate) == [{'Color':(255,0,0), 'Location':(3,4)}] 
+        assert bqs.query([validQuery, doubleDist]) == [{'Color':(255,0,0), 'Location':(3,4)}] 
+