def setUp(self): self.loop = asyncio.new_event_loop() asyncio.set_event_loop(self.loop) self.leds = LedsMock() self.ears = EarsMock() self.sound = SoundMock() self.ci = ChoreographyInterpreter(self.leds, self.ears, self.sound)
def __init__(self, test, random):
super().__init__()
self.test = test
self.leds = LedsMock()
self.ears = EarsMock()
self.sound = SoundMock()
self.ci = ChoreographyInterpreter(self.leds, self.ears, self.sound)
self.ci.taichi_random = random
class TestStreamingChoregraphy(unittest.TestCase):
def setUp(self):
self.loop = asyncio.new_event_loop()
asyncio.set_event_loop(self.loop)
self.leds = LedsMock()
self.ears = EarsMock()
self.sound = SoundMock()
self.ci = ChoreographyInterpreter(self.leds, self.ears, self.sound)
def test_streaming(self):
task = self.loop.create_task(self.ci.start(ChoreographyInterpreter.STREAMING_URN))
self.loop.run_until_complete(task)
task = self.loop.create_task(asyncio.sleep(1))
self.loop.run_until_complete(task)
self.assertEqual(self.sound.called_list, [])
self.assertTrue(len(self.leds.called_list) > 0)
self.assertTrue(len(self.ears.called_list) > 0)
task = self.loop.create_task(self.ci.stop())
self.loop.run_until_complete(task)
# Check at least some leds were on.
color_leds = 0
off_leds = 0
for ledi in self.leds.called_list:
if ledi.endswith('0,0,0)'):
off_leds=off_leds+1
else:
color_leds=color_leds+1
self.assertTrue(color_leds > 0)
self.assertTrue(off_leds > 0)
def test_streaming_n(self):
task = self.loop.create_task(self.ci.start(ChoreographyInterpreter.STREAMING_URN + ':3'))
self.loop.run_until_complete(task)
task = self.loop.create_task(asyncio.sleep(1))
self.loop.run_until_complete(task)
self.assertEqual(self.sound.called_list, [])
self.assertTrue(len(self.leds.called_list) > 0)
self.assertTrue(len(self.ears.called_list) > 0)
task = self.loop.create_task(self.ci.stop())
self.loop.run_until_complete(task)
def setUp(self):
self.leds = LedsMock()
self.ears = EarsMock()
self.sound = SoundMock()
self.ci = ChoreographyInterpreter(self.leds, self.ears, self.sound)
class TestChoreographyInterpreter(unittest.TestCase):
def setUp(self):
self.loop = asyncio.new_event_loop()
asyncio.set_event_loop(self.loop)
self.leds = LedsMock()
self.ears = EarsMock()
self.sound = SoundMock()
self.ci = ChoreographyInterpreter(self.leds, self.ears, self.sound)
def test_set_led_color(self):
chor = base64.b16decode("0007020304050607")
task = self.loop.create_task(self.ci.play_binary(chor))
self.loop.run_until_complete(task)
self.assertEqual(self.leds.called_list, ['set1(2,3,4,5)'])
self.assertEqual(self.ears.called_list, [])
self.assertEqual(self.sound.called_list, [])
def test_set_motor(self):
chor = base64.b16decode("0008010300")
task = self.loop.create_task(self.ci.play_binary(chor))
self.loop.run_until_complete(task)
self.assertEqual(self.leds.called_list, [])
self.assertEqual(self.ears.called_list, ['go(1,3,0)'])
self.assertEqual(self.sound.called_list, [])
def test_set_leds_color(self):
chor = base64.b16decode("0009020304")
task = self.loop.create_task(self.ci.play_binary(chor))
self.loop.run_until_complete(task)
self.assertEqual(self.leds.called_list, ['setall(2,3,4)'])
self.assertEqual(self.ears.called_list, [])
self.assertEqual(self.sound.called_list, [])
def test_set_led_off(self):
chor = base64.b16decode("000A02")
task = self.loop.create_task(self.ci.play_binary(chor))
self.loop.run_until_complete(task)
self.assertEqual(self.leds.called_list, ['set1(2,0,0,0)'])
self.assertEqual(self.ears.called_list, [])
self.assertEqual(self.sound.called_list, [])
def test_set_led_palette(self):
chor = base64.b16decode("000E0203")
task = self.loop.create_task(self.ci.play_binary(chor))
self.loop.run_until_complete(task)
self.assertEqual(self.leds.called_list, ['set1(2,0,0,0)'])
self.assertEqual(self.ears.called_list, [])
self.assertEqual(self.sound.called_list, [])
def test_randmidi(self):
chor = base64.b16decode("0010")
task = self.loop.create_task(self.ci.play_binary(chor))
self.loop.run_until_complete(task)
self.assertEqual(self.leds.called_list, [])
self.assertEqual(self.ears.called_list, [])
self.assertEqual(len(self.sound.called_list), 1)
soundcall = self.sound.called_list[0]
self.assertTrue(re.match(r'start\(.+\)', soundcall))
def test_avance(self):
chor = base64.b16decode("00110102")
task = self.loop.create_task(self.ci.play_binary(chor))
self.loop.run_until_complete(task)
self.assertEqual(self.leds.called_list, [])
self.assertEqual(self.ears.called_list, ['move(1,2,0)'])
self.assertEqual(self.sound.called_list, [])
def test_setmotordir(self):
chor = base64.b16decode("0014010100110102")
task = self.loop.create_task(self.ci.play_binary(chor))
self.loop.run_until_complete(task)
self.assertEqual(self.leds.called_list, [])
self.assertEqual(self.ears.called_list, ['move(1,-2,1)'])
self.assertEqual(self.sound.called_list, [])
def test_ifne(self):
chor = base64.b16decode("0012000000000A02")
task = self.loop.create_task(self.ci.play_binary(chor))
self.loop.run_until_complete(task)
self.assertEqual(self.leds.called_list, ['set1(2,0,0,0)'])
self.assertEqual(self.ears.called_list, [])
self.assertEqual(self.sound.called_list, [])
def test_rfidok(self):
chor = base64.b16decode(
"0101010100010E0007030000FF000000" +
"07020000FF00000007010000FF000001" +
"07030000000000000702000000000000" +
"070100000000000107040000FF000002" +
"070400000000000107040000FF000002" +
"0A04")
task = self.loop.create_task(self.ci.play_binary(chor))
self.loop.run_until_complete(task)
self.assertEqual(self.leds.called_list, [
'set1(3,0,0,255)',
'set1(2,0,0,255)',
'set1(1,0,0,255)',
'set1(3,0,0,0)',
'set1(2,0,0,0)',
'set1(1,0,0,0)',
'set1(4,0,0,255)',
'set1(4,0,0,0)',
'set1(4,0,0,255)',
'set1(4,0,0,0)'])
self.assertEqual(self.ears.called_list, [])
self.assertEqual(self.sound.called_list, [])