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_quarter(self): self.assertEqual(self.helper.round_up_to_quarter(0.99), 1) self.assertEqual(self.helper.round_up_to_quarter(0), 0) self.assertEqual(self.helper.round_up_to_quarter(0.1), 0.25) def test_round_down_to_quarter(self): self.assertEqual(self.helper.round_down_to_quarter(0.99), 0.75) self.assertEqual(self.helper.round_down_to_quarter(1.01), 1) self.assertEqual(self.helper.round_down_to_quarter(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 get_foil_price(self, api, product_id, language_id): # NOTE: This is a rough algorithm, designed to be safe and not to sell aggressively. # 1) See filter parameters below. # 2) Set price to lowest + (median - lowest / 4), rounded to closest quarter Euro. # 3) Undercut price in own country if not contradicting 2) # 4) Never go below 0.25 for foils account = api.get_account()['account'] articles = api.get_articles( product_id, **{ 'isFoil': 'true', 'isAltered': 'false', 'isSigned': 'false', 'minCondition': 'GD', 'idLanguage': language_id }) keys = ['idArticle', 'count', 'price', 'condition', 'seller'] foil_articles = [{x: y for x, y in art.items() if x in keys} for art in articles] local_articles = [] for article in foil_articles: if article['seller']['address']['country'] == account[ 'country'] and article['seller']['username'] != account[ 'username']: local_articles.append(article) local_table_data = [] for article in local_articles: if article: local_table_data.append([ article['seller']['username'], article['seller']['address']['country'], article['condition'], article['count'], article['price'] ]) table_data = [] for article in foil_articles: if article: table_data.append([ article['seller']['username'], article['seller']['address']['country'], article['condition'], article['count'], article['price'] ]) median_price = PyMkmHelper.calculate_median(table_data, 3, 4) lowest_price = PyMkmHelper.calculate_lowest(table_data, 4) median_guided = PyMkmHelper.round_up_to_quarter( lowest_price + (median_price - lowest_price) / 4) if len(local_table_data) > 0: # Undercut if there is local competition lowest_in_country = PyMkmHelper.round_down_to_quarter( PyMkmHelper.calculate_lowest(local_table_data, 4)) return max(0.25, min(median_guided, lowest_in_country - 0.25)) else: # No competition in our country, set price a bit higher. return PyMkmHelper.round_up_to_quarter(median_guided * 1.2)