diff options
Diffstat (limited to 'tests/TestBQS.py')
| -rw-r--r-- | tests/TestBQS.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/TestBQS.py b/tests/TestBQS.py index 7316c31..c46b917 100644 --- a/tests/TestBQS.py +++ b/tests/TestBQS.py @@ -13,13 +13,16 @@ class TestBQS(unittest.TestCase): 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 @@ -30,7 +33,6 @@ class TestBQS(unittest.TestCase): 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 |