import unittest

from utils import run_file


class TestPower(unittest.TestCase):
    def test_po_power(self):
        self.power("PO")

    def test_cs_power(self):
        self.power("CO")

    def power(self, mode):
        self.assertTrue(run_file(["power.alc", "primitives.alc"],
            ['1', '0', '2', '1', '5', '0', '2', '2', '3', '2', '10', '2', '10', '10'],
            ["1", "2", "1", "4", "9", "100", "10000000000"], mode))