Skip to content
Snippets Groups Projects
Select Git revision
  • 588806352b6029a55c0b223b446eb95f0c35b6cb
  • main default protected
2 results

get_spectrogram.py

Blame
    • test_configuration.py 2.63 KiB
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    44
    45
    46
    47
    48
    49
    50
    51
    52
    53
    54
    55
    56
    57
    58
    59
    60
    61
    62
    63
    64
    65
    66
    67
    68
    69
    70
    71
    72 
    

    

    

    import os
import unittest

import numpy as np

from multiview_platform.mono_multi_view_classifiers.utils import configuration

class Test_get_the_args(unittest.TestCase):

    def setUp(self):
        self.path_to_config_file = "multiview_platform/Tests/tmp_tests/config_temp.ini"
        os.mkdir("multiview_platform/Tests/tmp_tests")
        config_file = open(self.path_to_config_file, "w")
        config_file.write("[Base]\nfirst_arg = int ; 10\nsecond_arg = list_float ; 12.5 1e-06\n[Classification]\nthird_arg = bool ; yes")
        config_file.close()

    def tearDown(self):
        os.remove("multiview_platform/Tests/tmp_tests/config_temp.ini")
        os.rmdir("multiview_platform/Tests/tmp_tests")

    def test_file_loading(self):
        config_dict = configuration.get_the_args(self.path_to_config_file)
        self.assertEqual(type(config_dict), dict)

    def test_dict_format(self):
        config_dict = configuration.get_the_args(self.path_to_config_file)
        self.assertIn("Base", config_dict)
        self.assertIn("Classification", config_dict)
        self.assertIn("first_arg", config_dict["Base"])
        self.assertIn("third_arg", config_dict["Classification"])

    def test_arguments(self):
        config_dict = configuration.get_the_args(self.path_to_config_file)
        self.assertEqual(config_dict["Base"]["first_arg"], 10)
        self.assertEqual(config_dict["Base"]["second_arg"], [12.5, 1e-06])
        self.assertEqual(config_dict["Classification"]["third_arg"], True)

class Test_format_the_args(unittest.TestCase):

    def test_bool(self):
        value = configuration.format_raw_arg("bool ; yes")
        self.assertEqual(value, True)

    def test_int(self):
        value = configuration.format_raw_arg("int ; 1")
        self.assertEqual(value, 1)

    def test_float(self):
        value = configuration.format_raw_arg("float ; 1.5")
        self.assertEqual(value, 1.5)

    def test_string(self):
        value = configuration.format_raw_arg("str ; chicken_is_heaven")
        self.assertEqual(value, "chicken_is_heaven")

    def test_list_bool(self):
        value = configuration.format_raw_arg("list_bool ; yes no yes yes")
        self.assertEqual(value, [True, False, True, True])

    def test_list_int(self):
        value = configuration.format_raw_arg("list_int ; 1 2 3 4")
        self.assertEqual(value, [1,2,3,4])

    def test_list_float(self):
        value = configuration.format_raw_arg("list_float ; 1.5 1.6 1.7")
        self.assertEqual(value, [1.5, 1.6, 1.7])

    def test_list_string(self):
        value = configuration.format_raw_arg("list_str ; list string")
        self.assertEqual(value, ["list", "string"])