executorlib/tests/unit/standalone/test_inputcheck.py at 1998dd4892dfd69618199a4307fd88579ea8e4ff · pyiron/executorlib · GitHub

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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
import unittest

from executorlib.standalone.inputcheck import (
    check_wait_on_shutdown,
    check_command_line_argument_lst,
    check_gpus_per_worker,
    check_oversubscribe,
    check_executor,
    check_init_function,
    check_nested_flux_executor,
    check_flux_log_files,
    check_pmi,
    check_plot_dependency_graph,
    check_refresh_rate,
    check_resource_dict,
    check_resource_dict_is_empty,
    check_restart_limit,
    check_pmi_mode,
    check_max_workers_and_cores,
    check_hostname_localhost,
    check_pysqa_config_directory,
    check_file_exists,
    check_log_obj_size,
    validate_number_of_cores,
)


class TestInputCheck(unittest.TestCase):
    def test_check_wait_on_shutdown(self):
        with self.assertRaises(ValueError):
            check_wait_on_shutdown(wait_on_shutdown=False)

    def test_check_command_line_argument_lst(self):
        with self.assertRaises(ValueError):
            check_command_line_argument_lst(command_line_argument_lst=["a"])

    def test_check_gpus_per_worker(self):
        with self.assertRaises(TypeError):
            check_gpus_per_worker(gpus_per_worker=1)

    def test_check_oversubscribe(self):
        with self.assertRaises(ValueError):
            check_oversubscribe(oversubscribe=True)

    def test_check_executor(self):
        with self.assertRaises(ValueError):
            check_executor(executor=1)

    def test_check_init_function(self):
        with self.assertRaises(ValueError):
            check_init_function(init_function=1, block_allocation=False)

    def test_check_refresh_rate(self):
        with self.assertRaises(ValueError):
            check_refresh_rate(refresh_rate=1)

    def test_check_resource_dict(self):
        def simple_function(resource_dict):
            return resource_dict

        with self.assertRaises(ValueError):
            check_resource_dict(function=simple_function)

    def test_check_resource_dict_is_empty(self):
        with self.assertRaises(ValueError):
            check_resource_dict_is_empty(resource_dict={"a": 1})

    def test_check_pmi(self):
        with self.assertRaises(ValueError):
            check_pmi(backend="test", pmi="test")
        with self.assertRaises(ValueError):
            check_pmi(backend="flux_allocation", pmi="test")

    def test_check_restart_limit(self):
        with self.assertRaises(ValueError):
            check_restart_limit(restart_limit=1, block_allocation=False)

    def test_check_nested_flux_executor(self):
        with self.assertRaises(ValueError):
            check_nested_flux_executor(nested_flux_executor=True)

    def test_check_flux_log_files(self):
        with self.assertRaises(ValueError):
            check_flux_log_files(flux_log_files=True)

    def test_check_plot_dependency_graph(self):
        with self.assertRaises(ValueError):
            check_plot_dependency_graph(plot_dependency_graph=True)

    def test_check_pmi_mode(self):
        with self.assertRaises(ValueError):
            check_pmi_mode(pmi_mode="test")

    def test_check_max_workers_and_cores(self):
        with self.assertRaises(ValueError):
            check_max_workers_and_cores(max_workers=2, max_cores=None)
        with self.assertRaises(ValueError):
            check_max_workers_and_cores(max_workers=None, max_cores=2)
        with self.assertRaises(ValueError):
            check_max_workers_and_cores(max_workers=2, max_cores=2)

    def test_check_hostname_localhost(self):
        with self.assertRaises(ValueError):
            check_hostname_localhost(hostname_localhost=True)
        with self.assertRaises(ValueError):
            check_hostname_localhost(hostname_localhost=False)

    def test_check_pysqa_config_directory(self):
        with self.assertRaises(ValueError):
            check_pysqa_config_directory(pysqa_config_directory="path/to/config")

    def test_check_file_exists(self):
        with self.assertRaises(ValueError):
            check_file_exists(file_name=None)
        with self.assertRaises(ValueError):
            check_file_exists(file_name="/path/does/not/exist")

    def test_validate_number_of_cores(self):
        with self.assertRaises(ValueError):
            validate_number_of_cores(
                max_cores=None, max_workers=None, cores_per_worker=None
            )
        self.assertIsInstance(
            validate_number_of_cores(max_cores=1, max_workers=None, cores_per_worker=1),
            int,
        )
        self.assertIsInstance(
            validate_number_of_cores(
                max_cores=None, max_workers=1, cores_per_worker=None
            ),
            int,
        )

    def test_check_log_obj_size(self):
        with self.assertRaises(ValueError):
            check_log_obj_size(log_obj_size=True)