def initialize(display: Display, led: Led, speaker: Speaker,
dial: Dial) -> int:
"""Initialize the movment of the ball
:param display: Display module
:param led: Led module
:param speaker: Speaker module
:param dial: Dial module
:return: Score
"""
ball_pos = (20, 30)
ball_vel = (1, -1)
led.rgb = 0, 50, 0
score = 0
while True:
bar_pos = int(50 * dial.degree / 100)
ball_pos, ball_vel = update_screen(ball_pos, ball_vel, bar_pos,
display)
time.sleep(0.02)
if ball_pos[1] > 55 and (ball_pos[0] > bar_pos + 10
or ball_pos[0] < bar_pos - 10):
led.rgb = 50, 0, 0
break
elif ball_pos[1] > 55:
speaker.tune = 700, 100
time.sleep(0.1)
speaker.volume = 0
score += 1
display.clear()
return score
def setUp(self):
"""Set up test fixtures, if any."""
self.mock_kwargs = {"id_": -1, "uuid": -1, "msg_send_q": None}
self.display = Display(**self.mock_kwargs)
def eval_set_property(id, property_type, data, property_data_type):
eval_result = {
self.display.PropertyType.TEXT.value: [property_type],
self.display.PropertyType.VARIABLE.value: property_type,
self.display.PropertyType.CLEAR.value: property_type,
}.get(property_type)
return eval_result
self.display._set_property = mock.Mock(side_effect=eval_set_property)
self.display._msg_send_q = mock.Mock()
def setUp(self):
"""Set up test fixtures, if any."""
mock_input_values = (-1, -1, None)
mock_modules = [
SetupModule(*mock_input_values),
InputModule(*mock_input_values),
OutputModule(*mock_input_values),
Button(*mock_input_values),
Dial(*mock_input_values),
Env(*mock_input_values),
Gyro(*mock_input_values),
Ir(*mock_input_values),
Mic(*mock_input_values),
Ultrasonic(*mock_input_values),
Display(*mock_input_values),
Led(*mock_input_values),
Motor(*mock_input_values),
Speaker(*mock_input_values),
]
self.modi = MODI(nb_modules=len(mock_modules), test=True)
self.modi._ser_proc = mock.Mock()
self.modi._exe_thrd = mock.Mock()
self.modi._modules = mock_modules
class TestDisplay(unittest.TestCase):
"""Tests for 'Display' class."""
def setUp(self):
"""Set up test fixtures, if any."""
self.conn = MockConn()
self.mock_kwargs = [-1, -1, self.conn]
self.display = Display(*self.mock_kwargs)
def tearDown(self):
"""Tear down test fixtures, if any."""
del self.display
def test_set_text(self):
"""Test set_text method."""
mock_text = "abcd"
self.display.text = mock_text
clear_message = parse_message(0x04, 21, -1, (0, 0))
text_message = parse_message(
0x04, 17, -1, parse_data(mock_text + '\0', 'string')
)
self.assertEqual(
self.conn.send_list[0],
clear_message
)
self.assertEqual(
self.conn.send_list[1],
text_message
)
def test_show_variable(self):
"""Test set_variable method."""
mock_variable = 123
mock_position = 5
self.display.show_variable(mock_variable, mock_position, mock_position)
self.assertEqual(
self.conn.send_list[0],
parse_message(0x04, 22, -1, parse_data(
(mock_variable, mock_position, mock_position), 'display_var'
))
)
def test_clear(self):
"""Test clear method."""
self.display.clear()
clear_message = parse_message(0x04, 21, -1, (0, 0))
self.assertEqual(self.conn.send_list[0], clear_message)
class TestDisplay(unittest.TestCase):
"""Tests for 'Display' class."""
def setUp(self):
"""Set up test fixtures, if any."""
self.send_q = Queue()
self.mock_kwargs = {"id_": -1, "uuid": -1, "msg_send_q": self.send_q}
self.display = Display(**self.mock_kwargs)
def tearDown(self):
"""Tear down test fixtures, if any."""
del self.display
def test_set_text(self):
"""Test set_text method."""
mock_text = "abcd"
self.display.text = mock_text
clear_message = parse_message(0x04, 21, -1, (0, 0))
text_message = parse_message(0x04, 17, -1,
parse_data(mock_text, 'string'))
self.assertEqual(self.send_q.get(), clear_message)
self.assertEqual(self.send_q.get(), text_message)
def test_show_variable(self):
"""Test set_variable method."""
mock_variable = 123
mock_position = 5
self.display.show_variable(mock_variable, mock_position, mock_position)
self.assertEqual(
self.send_q.get(),
parse_message(
0x04, 22, -1,
parse_data((mock_variable, mock_position, mock_position),
'display_var')))
def test_clear(self):
"""Test clear method."""
self.display.clear()
clear_message = parse_message(0x04, 21, -1, (0, 0))
self.assertEqual(self.send_q.get(), clear_message)
def update_screen(pos: Tuple[int, int], vel: Tuple[int, int], bar: int,
display: Display) -> Tuple[Tuple[int, int], Tuple[int, int]]:
"""Update the screen of the display module after moving the ball
:param pos: Position of the ball
:param vel: Velocity of the ball
:param bar: Position of the bar
:param display: Reference to display module
:return: Tuple[Tuple[int, int], Tuple[int, int]]
"""
display.show_variable(0, pos[0], pos[1])
display.show_variable(1, bar, 60)
pos = (pos[0] + vel[0], pos[1] + vel[1])
if pos[0] < 0 or pos[0] > 40:
vel = (-vel[0], vel[1])
if pos[1] < 0 or pos[1] > 55:
vel = (vel[0], -vel[1])
if pos[1] < 0:
pos = (pos[0], 0)
if pos[0] < 0:
pos = (0, pos[1])
return pos, vel
def setUp(self):
"""Set up test fixtures, if any."""
self.send_q = Queue()
self.mock_kwargs = {"id_": -1, "uuid": -1, "msg_send_q": self.send_q}
self.display = Display(**self.mock_kwargs)
def setUp(self):
"""Set up test fixtures, if any."""
self.conn = MockConn()
self.mock_kwargs = [-1, -1, self.conn]
self.display = Display(*self.mock_kwargs)
class TestDisplay(unittest.TestCase):
"""Tests for 'Display' class."""
def setUp(self):
"""Set up test fixtures, if any."""
self.mock_kwargs = {"id_": -1, "uuid": -1, "msg_send_q": None}
self.display = Display(**self.mock_kwargs)
def eval_set_property(id, property_type, data, property_data_type):
eval_result = {
self.display.PropertyType.TEXT.value: [property_type],
self.display.PropertyType.VARIABLE.value: property_type,
self.display.PropertyType.CLEAR.value: property_type,
}.get(property_type)
return eval_result
self.display._set_property = mock.Mock(side_effect=eval_set_property)
self.display._msg_send_q = mock.Mock()
def tearDown(self):
"""Tear down test fixtures, if any."""
del self.display
def test_set_text(self):
"""Test set_text method."""
mock_text = "abcd"
self.display.set_text(text=mock_text)
expected_clear_params = (
self.mock_kwargs["id_"],
self.display.PropertyType.CLEAR.value,
bytes(2),
self.display.PropertyDataType.RAW,
)
expected_text_params = (
self.mock_kwargs["id_"],
self.display.PropertyType.TEXT.value,
mock_text,
self.display.PropertyDataType.STRING,
)
self.assertEqual(self.display._set_property.call_count, 2)
# TODO: Refactor two functions calls below to use assert_has_calls()
self.assertEqual(
mock.call(*expected_clear_params),
self.display._set_property.call_args_list[0],
)
self.assertEqual(
mock.call(*expected_text_params),
self.display._set_property.call_args_list[1],
)
def test_set_variable(self):
"""Test set_variable method."""
mock_variable = "12345"
mock_position = 5
self.display.set_variable(mock_variable, mock_position, mock_position)
expected_clear_params = (
self.mock_kwargs["id_"],
self.display.PropertyType.CLEAR.value,
bytes(2),
self.display.PropertyDataType.RAW,
)
expected_variable_params = (
self.mock_kwargs["id_"],
self.display.PropertyType.VARIABLE.value,
(mock_variable, mock_position, mock_position),
self.display.PropertyDataType.DISPLAY_VAR,
)
self.assertEqual(self.display._set_property.call_count, 2)
# TODO: Refactor two functions calls below to use assert_has_calls()
self.assertEqual(
mock.call(*expected_clear_params),
self.display._set_property.call_args_list[0],
)
self.assertEqual(
mock.call(*expected_variable_params),
self.display._set_property.call_args_list[1],
)
def test_clear(self):
"""Test clear method."""
self.display.clear()
# Check if set_property is called once with the specified arguments
expected_clear_params = (
self.mock_kwargs["id_"],
self.display.PropertyType.CLEAR.value,
bytes(2),
self.display.PropertyDataType.RAW,
)
self.display._set_property.assert_called_once_with(
*expected_clear_params)
# Check if correct message has been passed to serial_write_q
self.display._msg_send_q.put.assert_called_once_with(
self.display.PropertyType.CLEAR.value
)
