elite-engineering/tests/test_trade.py

import fractions
import typing
import unittest

from elite_engineering import trade


class TradeCalculationTests(unittest.TestCase):
    def format_ratio(self, ratio: typing.Optional[fractions.Fraction]) -> str:
        if not ratio:
            return "-"
        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 = [
            [
                self.format_ratio(
                    trade.ratio(grade_in, grade_out, across_categories=False)
                )
                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 = [
            [
                self.format_ratio(
                    trade.ratio(grade_in, grade_out, across_categories=True)
                )
                for grade_in in [1, 2, 3, 4, 5]
            ]
            for grade_out in [1, 2, 3, 4, 5]
        ]

        self.assertEqual(expected, actual)