Skip to content

Commit 35620e7

Browse files
committed
Improve test coverage slightly
1 parent 310ac39 commit 35620e7

File tree

2 files changed

+9
-4
lines changed

2 files changed

+9
-4
lines changed

test/test_cli.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
import pathlib
21
from unittest.mock import patch
32

3+
import pytest
44
from click.testing import CliRunner
55

66
from xcengine.cli import cli
@@ -9,14 +9,16 @@
99
@patch("xcengine.core.ScriptCreator.__init__")
1010
@patch("xcengine.core.ScriptCreator.convert_notebook_to_script")
1111
@patch("subprocess.run")
12-
def test_make_script(run_mock, convert_mock, init_mock, tmp_path):
12+
@pytest.mark.parametrize("verbose_arg", [[], ["--verbose"]])
13+
def test_make_script(run_mock, convert_mock, init_mock, tmp_path, verbose_arg):
1314
nb_path = tmp_path / "foo.ipynb"
1415
nb_path.touch()
1516
output_dir = tmp_path / "bar"
1617
init_mock.return_value = None
1718
runner = CliRunner()
1819
result = runner.invoke(
1920
cli,
21+
verbose_arg +
2022
[
2123
"make-script",
2224
"--batch",
@@ -37,4 +39,6 @@ def test_make_script(run_mock, convert_mock, init_mock, tmp_path):
3739
"--from-saved",
3840
]
3941
)
42+
from xcengine.cli import logging
43+
assert logging.getLogger().getEffectiveLevel() == (logging.DEBUG if "--verbose" in verbose_arg else logging.WARNING)
4044
assert result.exit_code == 0

test/test_parameters.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -275,12 +275,14 @@ def test_parameters_read_env_arguments(notebook_parameters):
275275
}.items()
276276
}
277277
)
278-
assert notebook_parameters.read_params_from_env() == {
278+
expected = {
279279
"some_int": 42,
280280
"some_float": 3.14159,
281281
"some_string": "foo",
282282
"some_bool": False,
283283
}
284+
assert notebook_parameters.read_params_from_env() == expected
285+
assert notebook_parameters.read_params_combined([]) == expected
284286

285287

286288
def test_parameters_read_params_combined(notebook_parameters):
@@ -290,7 +292,6 @@ def test_parameters_read_params_combined(notebook_parameters):
290292
prefix + k: v
291293
for k, v in {
292294
"some_int": "42",
293-
"some_float": "3.14159",
294295
"some_string": "foo",
295296
"some_bool": "False",
296297
}.items()

0 commit comments

Comments
 (0)