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
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)
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
)