Commit 7db92f46 authored by David Hendriks's avatar David Hendriks
Browse files

added imports for test cases. Oops

parent daa3192b
Loading
Loading
Loading
Loading
+12 −5
Original line number Diff line number Diff line
@@ -50,6 +50,11 @@ from binarycpython.tests.test_ensemble import (
    test_load_ensemble,
    test_open_ensemble,
)
from binarycpython.tests.test_event_logging import (
    test_event_file_processing,
    test_event_parameter_descriptions,
    test_event_types,
)
from binarycpython.tests.test_functions import (
    test_bin_data,
    test_create_arg_string,
@@ -162,15 +167,17 @@ from binarycpython.tests.tests_population_extensions.test_slurm import (
    test_slurm_outfile,
    test_slurmID,
)
from binarycpython.tests.tests_population_extensions.test_source_file_sampling import (
    test_source_file_sampling_column_type,
    test_source_file_sampling_command_type,
)
from binarycpython.tests.tests_population_extensions.test_spacing_functions import (
    test_const,
)
from binarycpython.tests.tests_population_extensions.test_version_info import (
    test_parse_binary_c_version_info,
    test_return_binary_c_version_info,
)

# from binarycpython.tests.test_spacing_functions import *
# from binarycpython.tests.test_population_options_defaults import *
# from binarycpython.tests.test_hpc_functions import *


if __name__ == "__main__":
    unittest.main()
+3 −2
Original line number Diff line number Diff line
@@ -23,10 +23,9 @@ TMP_DIR = temp_dir("tests", "test_spacing_functions")

shutil.rmtree(TMP_DIR)
os.makedirs(TMP_DIR, exist_ok=True)
distribution_functions_pop = Population()


class test_spacing_functions(unittest.TestCase):
class test_const(unittest.TestCase):
    """
    Unit test for spacing functions
    """
@@ -40,6 +39,8 @@ class test_spacing_functions(unittest.TestCase):
        Unittest for function const
        """

        distribution_functions_pop = Population()

        const_return = distribution_functions_pop.const_linear(1, 10, 10)
        self.assertTrue(
            (const_return == np.linspace(1, 10, 10 + 1)).all(),