modelverse/performance/test_create_matrix.py

import unittest
import utils


class TestCreateMatrix(unittest.TestCase):
    def test_po_create_matrix(self):
        self.create_matrix("PO")

    def test_co_create_matrix(self):
        self.create_matrix("CO")

    def create_matrix(self, mode):
        print(
            utils.run_file_single_output(
                ["test_harness.alc", "matrix.alc",
                 "create_matrix.alc", "primitives.alc",
                 "random.alc"],
                [100, 0],
                mode))