diff --git a/tests/test_esmf_profiling.py b/tests/test_esmf_profiling.py index de579a9..402a2b2 100644 --- a/tests/test_esmf_profiling.py +++ b/tests/test_esmf_profiling.py @@ -1,8 +1,9 @@ import pytest from pathlib import Path -from om3utils.esmf_profiling import ESMFProfilingParser +bt2 = pytest.importorskip("bt2", reason="Requires babeltrace2 python bindings") +from om3utils.esmf_profiling import ESMFProfilingParser def test_read_missing_esmf_profiling_dir(): with pytest.raises(FileNotFoundError): diff --git a/tests/test_esmf_trace.py b/tests/test_esmf_trace.py index 78e3d2f..c9d2429 100644 --- a/tests/test_esmf_trace.py +++ b/tests/test_esmf_trace.py @@ -1,5 +1,7 @@ import pytest +bt2 = pytest.importorskip("bt2", reason="Requires babeltrace2 python bindings") + from om3utils.esmf_trace import SinglePETTimingNode, MultiPETTimingNode