51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
import fractions
|
|
import unittest
|
|
|
|
from elite_engineering import trade
|
|
|
|
|
|
class TradeCalculationTests(unittest.TestCase):
|
|
def format_ratio(self, ratio: fractions.Fraction) -> str:
|
|
return "{} → {}".format(ratio.numerator, ratio.denominator)
|
|
|
|
def test_same_category_exchange(self) -> None:
|
|
expected = [
|
|
["1 → 1", "1 → 3", "1 → 9", "1 → 27", "1 → 81"],
|
|
["6 → 1", "1 → 1", "1 → 3", "1 → 9", "1 → 27"],
|
|
["36 → 1", "6 → 1", "1 → 1", "1 → 3", "1 → 9"],
|
|
["216 → 1", "36 → 1", "6 → 1", "1 → 1", "1 → 3"],
|
|
["-", "216 → 1", "36 → 1", "6 → 1", "1 → 1"],
|
|
]
|
|
|
|
actual = [
|
|
[
|
|
trade.ratio(grade_in, grade_out, across_categories=False)
|
|
.map(self.format_ratio)
|
|
.value_or("-")
|
|
for grade_in in [1, 2, 3, 4, 5]
|
|
]
|
|
for grade_out in [1, 2, 3, 4, 5]
|
|
]
|
|
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_different_category_exchange(self) -> None:
|
|
expected = [
|
|
["6 → 1", "2 → 1", "2 → 3", "2 → 9", "2 → 27"],
|
|
["36 → 1", "6 → 1", "2 → 1", "2 → 3", "2 → 9"],
|
|
["216 → 1", "36 → 1", "6 → 1", "2 → 1", "2 → 3"],
|
|
["-", "216 → 1", "36 → 1", "6 → 1", "2 → 1"],
|
|
["-", "-", "216 → 1", "36 → 1", "6 → 1"],
|
|
]
|
|
|
|
actual = [
|
|
[
|
|
trade.ratio(grade_in, grade_out, across_categories=True)
|
|
.map(self.format_ratio)
|
|
.value_or("-")
|
|
for grade_in in [1, 2, 3, 4, 5]
|
|
]
|
|
for grade_out in [1, 2, 3, 4, 5]
|
|
]
|
|
|
|
self.assertEqual(expected, actual)
|