class TestPyMkmHelperFunctions(unittest.TestCase):
def setUp(self):
self.helper = PyMkmHelper()
def test_calculate_average(self):
table = [['Yxskaft', 'SE', 'NM', 1, 1.21],
['Frazze11', 'SE', 'NM', 3, 1.3],
['andli826', 'SE', 'NM', 2, 1.82]]
self.assertEqual(self.helper.calculate_average(table, 3, 4), 1.46)
def test_calculate_median(self):
table = [['Yxskaft', 'SE', 'NM', 1, 1.21],
['Frazze11', 'SE', 'NM', 3, 1.3],
['andli826', 'SE', 'NM', 2, 1.82]]
self.assertEqual(self.helper.calculate_median(table, 3, 4), 1.3)
self.assertEqual(self.helper.calculate_average(table, 3, 4), 1.46)
def test_calculate_lowest(self):
table = [['Yxskaft', 'SE', 'NM', 1, 1.21],
['Frazze11', 'SE', 'NM', 3, 1.3],
['andli826', 'SE', 'NM', 2, 1.82]]
self.assertEqual(self.helper.calculate_lowest(table, 4), 1.21)
def test_round_up_to_limit(self):
self.assertEqual(self.helper.round_up_to_limit(0.25, 0.99), 1)
self.assertEqual(self.helper.round_up_to_limit(0.25, 0), 0)
self.assertEqual(self.helper.round_up_to_limit(0.25, 0.1), 0.25)
self.assertEqual(self.helper.round_up_to_limit(0.1, 0.99), 1)
self.assertEqual(self.helper.round_up_to_limit(0.01, 0.011), 0.02)
self.assertEqual(self.helper.round_up_to_limit(0.01, 1), 1)
self.assertEqual(self.helper.round_up_to_limit(1, 0.1), 1)
def test_round_down_to_limit(self):
self.assertEqual(self.helper.round_down_to_limit(0.25, 0.99), 0.75)
self.assertEqual(self.helper.round_down_to_limit(0.25, 1.01), 1)
self.assertEqual(self.helper.round_down_to_limit(0.25, 0.1), 0)
self.assertEqual(self.helper.round_down_to_limit(0.1, 0.99), 0.9)
self.assertEqual(self.helper.round_down_to_limit(0.01, 0.011), 0.01)
self.assertEqual(self.helper.round_down_to_limit(0.01, 1), 1)
self.assertEqual(self.helper.round_down_to_limit(1, 0.1), 0)
@patch('sys.stdout', new_callable=io.StringIO)
@patch('builtins.input', side_effect=['y', 'n', 'p', 'n'])
def test_prompt_bool(self, mock_input, mock_stdout):
self.assertTrue(self.helper.prompt_bool('test_y'))
self.assertFalse(self.helper.prompt_bool('test_n'))
self.helper.prompt_bool('test_error')
self.assertRegex(mock_stdout.getvalue(),
r'\nPlease answer with y\/n\n')
@patch('builtins.input', side_effect=['y'])
def test_prompt_string(self, mock_input):
self.assertEqual(self.helper.prompt_string('test'), 'y')
def print_product_top_list(self, title_string, table_data, sort_column, rows):
print(70*'-')
print('{} \n'.format(title_string))
print(tb.tabulate(sorted(table_data, key=lambda x: x[sort_column], reverse=False)[:rows],
headers=['Username', 'Country',
'Condition', 'Count', 'Price'],
tablefmt="simple"))
print(70*'-')
print('Total average price: {}, Total median price: {}, Total # of articles: {}\n'.format(
str(PyMkmHelper.calculate_average(table_data, 3, 4)),
str(PyMkmHelper.calculate_median(table_data, 3, 4)),
str(len(table_data))
)
)
