modelverse/performance/test_matrix_gauss_jordan.py

import unittest
import utils


class TestMatrixGaussJordan(unittest.TestCase):
    def test_interpreter_matrix_gauss_jordan(self):
        self.matrix_gauss_jordan(utils.OPTIMIZATION_LEVEL_INTERPRETER)

    def test_baseline_jit_matrix_gauss_jordan(self):
        self.matrix_gauss_jordan(utils.OPTIMIZATION_LEVEL_BASELINE_JIT)

    def matrix_gauss_jordan(self, optimization_level):
        utils.write_perf_to_file(
            'matrix_gauss_jordan', optimization_level,
            utils.run_perf_test(
                ["test_harness.alc", "matrix.alc",
                 "matrix_gauss_jordan.alc", "primitives.alc",
                 "random.alc", "jit.alc"],
                [25],
                optimization_level))