88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
import fractions
|
|
import unittest
|
|
|
|
from returns.curry import partial
|
|
from returns.maybe import Maybe
|
|
|
|
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"],
|
|
]
|
|
|
|
material = partial(
|
|
trade.Material,
|
|
"name",
|
|
trade.MaterialType.raw,
|
|
"category",
|
|
)
|
|
actual = [
|
|
[
|
|
material(grade_in)
|
|
.trade_ratio(material(grade_out))
|
|
.map(self.format_ratio)
|
|
.value_or("-")
|
|
for grade_out in [1, 2, 3, 4, 5]
|
|
]
|
|
for grade_in 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"],
|
|
]
|
|
|
|
type_ = trade.MaterialType.raw
|
|
material = partial(trade.Material, "name", trade.MaterialType.raw)
|
|
actual = [
|
|
[
|
|
material("category", grade_in)
|
|
.trade_ratio(material("other", grade_out))
|
|
.map(self.format_ratio)
|
|
.value_or("-")
|
|
for grade_out in [1, 2, 3, 4, 5]
|
|
]
|
|
for grade_in in [1, 2, 3, 4, 5]
|
|
]
|
|
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_different_type_exchange(self) -> None:
|
|
expected = [
|
|
["-", "-", "-", "-", "-"],
|
|
["-", "-", "-", "-", "-"],
|
|
["-", "-", "-", "-", "-"],
|
|
["-", "-", "-", "-", "-"],
|
|
["-", "-", "-", "-", "-"],
|
|
]
|
|
|
|
type_ = trade.MaterialType.raw
|
|
material = partial(trade.Material, "name")
|
|
actual = [
|
|
[
|
|
material(trade.MaterialType.raw, "category", grade_in)
|
|
.trade_ratio(material(trade.MaterialType.encoded, "other", grade_out))
|
|
.map(self.format_ratio)
|
|
.value_or("-")
|
|
for grade_out in [1, 2, 3, 4, 5]
|
|
]
|
|
for grade_in in [1, 2, 3, 4, 5]
|
|
]
|
|
|
|
self.assertEqual(expected, actual)
|