class SmartHomeTest(unittest.TestCase):
def setUp(self):
self.devices = {}
self.atomese = Atomese()
self.atomese.add_token(
"dev:[^\\s)]+",
lambda token: ValueAtom(self._get_device(token[4:])))
def test_turn_lamps_on_via_grounded_match(self):
kb = GroundingSpace()
kb.add_atom(
E(S("isa"), ValueAtom(self._get_device("bedroom-lamp")),
S("lamp")))
kb.add_atom(
E(S("isa"), ValueAtom(self._get_device("kitchen-lamp")),
S("lamp")))
target = GroundingSpace()
target.add_atom(
E(MatchAtom(), ValueAtom(kb), E(S("isa"), V("x"), S("lamp")),
E(S("q"), CallAtom("turn_on"), V("x"))))
interpret_until_result(target, GroundingSpace())
self.assertTrue(self._get_device("kitchen-lamp").is_on)
self.assertTrue(self._get_device("bedroom-lamp").is_on)
def test_turn_lamps_on_via_grounded_match_atomese(self):
Logger.setLevel(Logger.DEBUG)
kb = self.atomese.parse('''
(isa dev:kitchen-lamp lamp)
(isa dev:bedroom-lamp lamp)
''')
self.atomese.add_atom("kb", ValueAtom(kb))
target = self.atomese.parse('''
(match kb (isa $x lamp) (q call:turn_on $x))
''')
interpret_until_result(target, kb)
self.assertTrue(self._get_device("kitchen-lamp").is_on)
self.assertTrue(self._get_device("bedroom-lamp").is_on)
def test_turn_lamps_on_via_interpreter_matching(self):
kb = self.atomese.parse('''
(= (lamp) dev:kitchen-lamp)
(= (lamp) dev:bedroom-lamp)
''')
target = self.atomese.parse('''
(call:turn_on (lamp))
''')
interpret_until_result(target, kb)
self.assertTrue(self._get_device("kitchen-lamp").is_on)
self.assertTrue(self._get_device("bedroom-lamp").is_on)
def test_turn_lamps_on_via_condition_and_matching(self):
kb = self.atomese.parse('''
(= (if True $then $else) $then)
(= (if False $then $else) $else)
(= (lamp dev:kitchen-lamp) True)
(= (lamp dev:bedroom-lamp) True)
(= (lamp dev:kettle) False)
(= (turn_lamp_on) (if (lamp $x) (call:turn_on $x) nop))
''')
target = self.atomese.parse('''
(turn_lamp_on)
''')
interpret_until_result(target, kb)
interpret_until_result(target, kb)
interpret_until_result(target, kb)
self.assertTrue(self._get_device("kitchen-lamp").is_on)
self.assertTrue(self._get_device("bedroom-lamp").is_on)
self.assertFalse(self._get_device("toilet").is_on)
def _get_device(self, name):
if not name in self.devices:
self.devices[name] = Device(name)
return self.devices[name]