test_basics.py

# SPDX-FileCopyrightText: 2021 Gabriel J. Schwarzkopf <sispo-devs@outlook.com>
#
# SPDX-License-Identifier: GPL-3.0-or-later

"""Test suite."""

import unittest
from pathlib import Path

import numpy as np
from sispo.sim import utils


class TestUtils(unittest.TestCase):
    """Basic unit tests"""

    def test_check_dir(self):
        dir_name = "dir_test"

        file_dir = Path(__file__).parent.resolve()
        test_dir = file_dir / dir_name

        dir = utils.check_dir(test_dir)
        self.assertTrue(test_dir == dir)
        self.assertTrue(test_dir.exists())

        test_dir_relative = test_dir / ".." / dir_name
        self.assertTrue(test_dir == utils.check_dir(test_dir_relative))
        self.assertTrue(test_dir.exists())

        Path.rmdir(test_dir)

    def test_write_vec_string(self):
        vec = (1.234, 2.345, 3.456)

        self.assertEqual(utils.write_vec_string(vec, 0), "[1,2,3]")
        self.assertEqual(utils.write_vec_string(vec, 1), "[1.2,2.3,3.5]")
        self.assertEqual(utils.write_vec_string(vec, 2), "[1.23,2.35,3.46]")
        self.assertEqual(utils.write_vec_string(vec, 3), "[1.234,2.345,3.456]")
        self.assertEqual(utils.write_vec_string(vec, 4),
                         "[1.2340,2.3450,3.4560]")

    def test_write_mat_string(self):
        mat = ((1.234, 2.345, 3.456),
               (4.567, 5.678, 6.789), (7.891, 8.912, 9.123))

        self.assertEqual(utils.write_mat_string(mat, 0),
                         "[[1,2,3],[5,6,7],[8,9,9]]")
        self.assertEqual(utils.write_mat_string(mat, 1),
                         "[[1.2,2.3,3.5],[4.6,5.7,6.8],[7.9,8.9,9.1]]")
        self.assertEqual(utils.write_mat_string(mat, 3),
                         "[[1.234,2.345,3.456],[4.567,5.678,6.789],[7.891,8.912,9.123]]")
        self.assertEqual(utils.write_mat_string(mat, 4),
                         "[[1.2340,2.3450,3.4560],[4.5670,5.6780,6.7890],[7.8910,8.9120,9.1230]]")

    def test_serialise(self):
        test_array = np.array([0, 1, 2, 3, 4, 5, 6])
        test_float = "1.234567"

        self.assertEqual(utils.serialise(test_array), [0, 1, 2, 3, 4, 5, 6])
        self.assertEqual(utils.serialise(test_float), float(test_float))


if __name__ == "__main__":
    unittest.main()