import unittest

from utils import run_file


class TestFactorial(unittest.TestCase):
    def test_po_factorial(self):
        self.factorial("PO")

    def test_co_factorial(self):
        self.factorial("CO")

    def factorial(self, mode):
        self.assertTrue(run_file(["factorial.alc", "primitives.alc"],
            [1, 2, 3, 4],
            [1, 2, 6, 24],
            mode))