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)