Merge pull request #3 from Krekep/master

EgorOrachyov · web-flow · commit 7bf439defbb4 · 2021-03-07T20:11:05.000+03:00
[Python] Wrapper tests
diff --git a/python/tests/test_duplicate.py b/python/tests/test_duplicate.py
@@ -0,0 +1,26 @@
+import unittest
+import test_utils
+import utils.io_file
+
+
+class TestMatrixDuplicate(unittest.TestCase):
+    def setUp(self) -> None:
+        self.input_matrices = [["matrix_1.mtx"], ["matrix_2.mtx"]]
+        self.result_matrices = ["duplicate_res_1.mtx", "duplicate_res_2.mtx"]
+
+    def test_duplicate(self):
+        """
+        Unit test for duplicate of matrix
+        """
+        for i in range(len(self.input_matrices)):
+            matrices = list()
+            for matrix in self.input_matrices[i]:
+                matrices.append(utils.io_file.build_matrix_by_name(matrix))
+            expected_matrix = utils.io_file.build_matrix_by_name(self.result_matrices[i])
+            actual_matrix = matrices[0].dup()
+
+            self.assertTrue(test_utils.compare_matrix(expected_matrix, actual_matrix))
+
+
+if __name__ == "__main__":
+    unittest.main()
diff --git a/python/tests/test_extract_matrix.py b/python/tests/test_extract_matrix.py
@@ -0,0 +1,27 @@
+import unittest
+import test_utils
+import utils.io_file
+
+
+class TestMatrixExtractMatrix(unittest.TestCase):
+    def setUp(self) -> None:
+        self.input_matrices = [["matrix_1.mtx"], ["matrix_2.mtx"]]
+        self.result_matrices = ["extract_res_1.mtx", "extract_res_2.mtx"]
+
+    def test_extract_matrix(self):
+        """
+        Unit test for extract submatrix from left-upper corner of matrix
+        """
+        for i in range(len(self.input_matrices)):
+            matrices = list()
+            for matrix in self.input_matrices[i]:
+                matrices.append(utils.io_file.build_matrix_by_name(matrix))
+            expected_matrix = utils.io_file.build_matrix_by_name(self.result_matrices[i])
+
+            actual_matrix = matrices[0].extract_matrix(0, 0, expected_matrix.shape)
+
+            self.assertTrue(test_utils.compare_matrix(expected_matrix, actual_matrix))
+
+
+if __name__ == "__main__":
+    unittest.main()
diff --git a/python/tests/test_kronecker.py b/python/tests/test_kronecker.py
@@ -0,0 +1,28 @@
+import unittest
+import test_utils
+import utils.io_file
+
+
+class TestMatrixKronecker(unittest.TestCase):
+    def setUp(self) -> None:
+        self.input_matrices = [["matrix_1.mtx", "matrix_2.mtx"], ["matrix_3.mtx", "matrix_4.mtx"]]
+        self.result_matrices = ["kronecker_res_12.mtx", "kronecker_res_34.mtx"]
+
+    def test_kronecker(self):
+        """
+        Unit test for kronecker product of two matrices
+        """
+        for i in range(len(self.input_matrices)):
+            matrices = list()
+            for matrix in self.input_matrices[i]:
+                matrices.append(utils.io_file.build_matrix_by_name(matrix))
+            expected_matrix = utils.io_file.build_matrix_by_name(self.result_matrices[i])
+            first_matrix = matrices[0]
+            second_matrix = matrices[1]
+            actual_matrix = first_matrix.kronecker(second_matrix)
+
+            self.assertTrue(test_utils.compare_matrix(expected_matrix, actual_matrix))
+
+
+if __name__ == "__main__":
+    unittest.main()
diff --git a/python/tests/test_matrix_add.py b/python/tests/test_matrix_add.py
@@ -0,0 +1,28 @@
+import unittest
+import test_utils
+import utils.io_file
+
+
+class TestMatrixAdd(unittest.TestCase):
+    def setUp(self) -> None:
+        self.input_matrices = [["matrix_1.mtx", "matrix_2.mtx"], ["matrix_3.mtx", "matrix_4.mtx"]]
+        self.result_matrices = ["add_res_12.mtx", "add_res_34.mtx"]
+
+    def test_add(self):
+        """
+        Unit test for addition of two matrices
+        """
+        for i in range(len(self.input_matrices)):
+            matrices = list()
+            for matrix in self.input_matrices[i]:
+                matrices.append(utils.io_file.build_matrix_by_name(matrix))
+            expected_matrix = utils.io_file.build_matrix_by_name(self.result_matrices[i])
+            first_matrix = matrices[0]
+            second_matrix = matrices[1]
+            actual_matrix = first_matrix.ewiseadd(second_matrix)
+
+            self.assertTrue(test_utils.compare_matrix(expected_matrix, actual_matrix))
+
+
+if __name__ == "__main__":
+    unittest.main()
diff --git a/python/tests/test_matrix_mxm.py b/python/tests/test_matrix_mxm.py
@@ -0,0 +1,28 @@
+import unittest
+import test_utils
+import utils.io_file
+
+
+class TestMatrixMxm(unittest.TestCase):
+    def setUp(self) -> None:
+        self.input_matrices = [["matrix_1.mtx", "matrix_2.mtx"], ["matrix_3.mtx", "matrix_4.mtx"]]
+        self.result_matrices = ["mxm_res_12.mtx", "mxm_res_34.mtx"]
+
+    def test_mxm(self):
+        """
+        Unit test for multiplication of two matrices
+        """
+        for i in range(len(self.input_matrices)):
+            matrices = list()
+            for matrix in self.input_matrices[i]:
+                matrices.append(utils.io_file.build_matrix_by_name(matrix))
+            expected_matrix = utils.io_file.build_matrix_by_name(self.result_matrices[i])
+            first_matrix = matrices[0]
+            second_matrix = matrices[1]
+            actual_matrix = first_matrix.mxm(second_matrix)
+
+            self.assertTrue(test_utils.compare_matrix(expected_matrix, actual_matrix))
+
+
+if __name__ == "__main__":
+    unittest.main()
diff --git a/python/tests/test_matrix_properties.py b/python/tests/test_matrix_properties.py
@@ -0,0 +1,45 @@
+import unittest
+import utils.io_file
+
+
+class TestMatrixMethods(unittest.TestCase):
+    def setUp(self) -> None:
+        self.input_matrices = [["matrix_1.mtx"], ["matrix_2.mtx"]]
+        self.result_matrices = ["property_res_1.mtx", "property_res_2.mtx"]
+
+        self.matrices = list()
+        self.result_property = list()
+        for i in range(len(self.input_matrices)):
+            for matrix in self.input_matrices[i]:
+                self.matrices.append(utils.io_file.build_matrix_by_name(matrix))
+
+        for i in range(len(self.result_matrices)):
+            self.result_property.append(utils.io_file.build_matrix_by_name(self.result_matrices[i]))
+
+    def tearDown(self) -> None:
+        """
+        Final actions
+        Performed AFTER tests
+        """
+
+        pass
+
+    def test_nrows(self):
+        for i in range(len(self.matrices)):
+            self.assertEqual(self.result_property[i].nrows, self.matrices[i].nrows)
+
+    def test_ncols(self):
+        for i in range(len(self.matrices)):
+            self.assertEqual(self.result_property[i].ncols, self.matrices[i].ncols)
+
+    def test_nvals(self):
+        for i in range(len(self.matrices)):
+            self.assertEqual(self.result_property[i].nvals, self.matrices[i].nvals)
+
+    def test_shape(self):
+        for i in range(len(self.matrices)):
+            self.assertEqual(self.result_property[i].shape, self.matrices[i].shape)
+
+
+if __name__ == "__main__":
+    unittest.main()
diff --git a/python/tests/test_reduce.py b/python/tests/test_reduce.py
@@ -0,0 +1,26 @@
+import unittest
+import test_utils
+import utils.io_file
+
+
+class TestMatrixReduce(unittest.TestCase):
+    def setUp(self) -> None:
+        self.input_matrices = [["matrix_1.mtx"], ["matrix_2.mtx"]]
+        self.result_matrices = ["reduce_res_1.mtx", "reduce_res_2.mtx"]
+
+    def test_reduce(self):
+        """
+        Unit test for reduce of matrix
+        """
+        for i in range(len(self.input_matrices)):
+            matrices = list()
+            for matrix in self.input_matrices[i]:
+                matrices.append(utils.io_file.build_matrix_by_name(matrix))
+            expected_matrix = utils.io_file.build_matrix_by_name(self.result_matrices[i])
+            actual_matrix = matrices[0].reduce()
+
+            self.assertTrue(test_utils.compare_matrix(expected_matrix, actual_matrix))
+
+
+if __name__ == "__main__":
+    unittest.main()
diff --git a/python/tests/test_to_lists.py b/python/tests/test_to_lists.py
@@ -0,0 +1,37 @@
+import unittest
+import utils.io_file
+
+
+class TestMatrixToLists(unittest.TestCase):
+    def setUp(self) -> None:
+        self.input_matrices = [["matrix_1.mtx"], ["matrix_2.mtx"]]
+        self.result_matrices = ["to_lists_res_1.mtx", "to_lists_res_2.mtx"]
+
+        self.matrices = list()
+        self.result_lists = list()
+        for i in range(len(self.input_matrices)):
+            for matrix in self.input_matrices[i]:
+                self.matrices.append(utils.io_file.build_matrix_by_name(matrix))
+
+        for i in range(len(self.result_matrices)):
+            self.result_lists.append(utils.io_file.build_matrix_by_name(self.result_matrices[i]))
+
+    def test_to_lists(self):
+        """
+        Unit test to extract a matrix as two lists
+        """
+        for i in range(len(self.matrices)):
+
+            actual_rows, actual_cols = self.matrices[i].to_lists()
+            expected_rows, expected_cols = self.result_lists[i].to_lists()
+
+            # element-wise compare
+            if len(actual_cols) != len(expected_cols) and len(expected_rows) != len(actual_rows):
+                self.assertTrue(False, msg="Length of lists doesn't match")
+            for j in range(len(actual_cols)):
+                self.assertEqual(expected_rows[j], actual_rows[j])
+                self.assertEqual(expected_cols[j], actual_cols[j])
+
+
+if __name__ == "__main__":
+    unittest.main()
diff --git a/python/tests/test_transpose.py b/python/tests/test_transpose.py
@@ -0,0 +1,26 @@
+import unittest
+import test_utils
+import utils.io_file
+
+
+class TestMatrixTranspose(unittest.TestCase):
+    def setUp(self) -> None:
+        self.input_matrices = [["matrix_1.mtx"], ["matrix_2.mtx"]]
+        self.result_matrices = ["transpose_res_1.mtx", "transpose_res_2.mtx"]
+
+    def test_transpose(self):
+        """
+        Unit test for transpose of matrix
+        """
+        for i in range(len(self.input_matrices)):
+            matrices = list()
+            for matrix in self.input_matrices[i]:
+                matrices.append(utils.io_file.build_matrix_by_name(matrix))
+            expected_matrix = utils.io_file.build_matrix_by_name(self.result_matrices[i])
+            actual_matrix = matrices[0].transpose()
+
+            self.assertTrue(test_utils.compare_matrix(expected_matrix, actual_matrix))
+
+
+if __name__ == "__main__":
+    unittest.main()
diff --git a/python/tests/test_utils.py b/python/tests/test_utils.py
@@ -0,0 +1,22 @@
+import pycubool
+
+
+def compare_matrix(a: pycubool.Matrix, b: pycubool.Matrix) -> bool:
+    a_rows, a_cols = a.to_lists()
+    b_rows, b_cols = b.to_lists()
+
+    if len(a_rows) != len(b_rows):
+        return False
+    n = len(a_rows)
+    for i in range(n):
+        if a_rows[i] != b_rows[i]:
+            return False
+
+    if len(a_cols) != len(b_cols):
+        return False
+    m = len(a_cols)
+    for i in range(m):
+        if a_cols[i] != b_cols[i]:
+            return False
+
+    return True
diff --git a/python/utils/add_generate.py b/python/utils/add_generate.py
@@ -0,0 +1,11 @@
+import io_file
+
+
+def add(first_name: str, second_name: str, res_name: str):
+    first_matrix = io_file.build_matrix_by_name(first_name)
+    second_matrix = io_file.build_matrix_by_name(second_name)
+    result_matrix = first_matrix.ewiseadd(second_matrix)
+
+    size = result_matrix.shape
+    rows, cols = result_matrix.to_lists()
+    io_file.write_matrix_to_file(size, rows, cols, res_name)
diff --git a/python/utils/duplicate_generate.py b/python/utils/duplicate_generate.py
@@ -0,0 +1,10 @@
+import io_file
+
+
+def duplicate(first_name: str, res_name: str):
+    first_matrix = io_file.build_matrix_by_name(first_name)
+    result_matrix = first_matrix.dup()
+
+    size = result_matrix.shape
+    rows, cols = result_matrix.to_lists()
+    io_file.write_matrix_to_file(size, rows, cols, res_name)
diff --git a/python/utils/extract_generate.py b/python/utils/extract_generate.py
@@ -0,0 +1,11 @@
+import io_file
+
+
+def extract(first_name: str, res_name: str):
+    first_matrix = io_file.build_matrix_by_name(first_name)
+
+    result_matrix = first_matrix.extract_matrix(0, 0, first_matrix.shape)
+
+    size = result_matrix.shape
+    rows, cols = result_matrix.to_lists()
+    io_file.write_matrix_to_file(size, rows, cols, res_name)
diff --git a/python/utils/io_file.py b/python/utils/io_file.py
@@ -0,0 +1,47 @@
+import os
+
+import pycubool
+
+__all__ = [
+    "build_matrix_by_name",
+    "write_matrix_to_file",
+    "path_to_matrices"
+]
+
+path_to_matrices = os.environ["TestMatrix_Path"]
+
+
+def read_matrix_mtx(path: str):
+    """
+    Read matrix from file
+    """
+    with open(path, 'r') as file:
+        n, m, nvals = map(int, file.readline().split())
+        rows = list()
+        cols = list()
+        for k in range(nvals):
+            i, j = map(int, file.readline().split())
+            rows.append(i)
+            cols.append(j)
+    return n, m, [rows, cols]
+
+
+def build_matrix_by_name(matrix_name: str) -> pycubool.Matrix:
+    n, m, matrix = read_matrix_mtx(path_to_matrices + matrix_name)
+    nvals = len(matrix[0])
+    result = pycubool.Matrix.empty([n, m])
+    result.build(matrix[0], matrix[1], nvals)
+
+    return result
+
+
+def write_matrix_to_file(size: list, rows: list, cols: list, name: str):
+    if len(rows) != len(cols):
+        raise Exception("Size of rows and cols arrays must match the nval values")
+
+    nvals = len(rows)
+    path = path_to_matrices + name
+    with open(path, "w") as file:
+        file.write(f"{size[0]} {size[1]} {nvals}\n")
+        for i in range(nvals):
+            file.write(f"{rows[i]} {cols[i]}\n")
diff --git a/python/utils/kronecker_generate.py b/python/utils/kronecker_generate.py
@@ -0,0 +1,11 @@
+import io_file
+
+
+def kronecker(first_name: str, second_name: str, res_name: str):
+    first_matrix = io_file.build_matrix_by_name(first_name)
+    second_matrix = io_file.build_matrix_by_name(second_name)
+    result_matrix = first_matrix.kronecker(second_matrix)
+
+    size = result_matrix.shape
+    rows, cols = result_matrix.to_lists()
+    io_file.write_matrix_to_file(size, rows, cols, res_name)
diff --git a/python/utils/matrix_generate.py b/python/utils/matrix_generate.py
diff --git a/python/utils/mxm_generate.py b/python/utils/mxm_generate.py
diff --git a/python/utils/properties_generate.py b/python/utils/properties_generate.py
diff --git a/python/utils/reduce_generate.py b/python/utils/reduce_generate.py
diff --git a/python/utils/to_lists_generate.py b/python/utils/to_lists_generate.py
diff --git a/python/utils/transpose_generate.py b/python/utils/transpose_generate.py
