def test_summary(self):
"""Check if status bar is updated when data is received"""
data = Table("iris")
input_sum = self.widget.info.set_input_summary = Mock()
output_sum = self.widget.info.set_output_summary = Mock()
self.send_signal(self.widget.Inputs.data, data)
input_sum.assert_called_with(int(StateInfo.format_number(len(data))))
output = self.get_output(self.widget.Outputs.data)
output_sum.assert_called_with(int(StateInfo.format_number(len(output))))
input_sum.reset_mock()
output_sum.reset_mock()
self.send_signal(self.widget.Inputs.data, None)
input_sum.assert_called_once()
self.assertEqual(input_sum.call_args[0][0].brief, "")
output_sum.assert_called_once()
self.assertEqual(output_sum.call_args[0][0].brief, "")
def test_format_number(self):
self.assertEqual(StateInfo.format_number(9999), "9999")
self.assertEqual(StateInfo.format_number(12_345), "12.3k")
self.assertEqual(StateInfo.format_number(12_000), "12k")
self.assertEqual(StateInfo.format_number(123_456), "123k")
self.assertEqual(StateInfo.format_number(99_999), "100k")
self.assertEqual(StateInfo.format_number(1_234_567), "1.23M")
self.assertEqual(StateInfo.format_number(999_999), "1M")
self.assertEqual(StateInfo.format_number(1_000_000), "1M")
def format_short(partial):
summary = partial.summary
if summary is None:
return "-"
if isinstance(summary, int):
return StateInfo.format_number(summary)
if isinstance(summary, str):
return summary
raise ValueError("summary must be None, string or int; "
f"got {type(summary).__name__}")