elite-engineering/tests/test_trade.py

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)