blob: 8dc90b20d7f9c418c26dc852aea6239ff165f829 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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) == []
|