fix tests
This commit is contained in:
19
tests/node_tests/conftest.py
Normal file
19
tests/node_tests/conftest.py
Normal file
@@ -0,0 +1,19 @@
|
||||
from pathlib import Path
|
||||
import pytest
|
||||
|
||||
_DIR = Path(__file__).parent
|
||||
|
||||
|
||||
def pytest_collect_file(parent, file_path):
|
||||
"""Collect all non-private .py files in this directory as test modules.
|
||||
|
||||
Allows test files to be named after their source module (e.g. acf_2d.py)
|
||||
rather than requiring a test_ prefix.
|
||||
"""
|
||||
if (
|
||||
file_path.parent == _DIR
|
||||
and file_path.suffix == ".py"
|
||||
and not file_path.name.startswith("_")
|
||||
and file_path.name != "conftest.py"
|
||||
):
|
||||
return pytest.Module.from_parent(parent, path=file_path)
|
||||
@@ -14,7 +14,7 @@ def test_threshold_mask():
|
||||
ThresholdMask._broadcast_fn = lambda nid, uri: previews.append(uri)
|
||||
ThresholdMask._current_node_id = "test"
|
||||
|
||||
mask, = node.process(field, method="absolute", threshold=0.5, direction="above")
|
||||
mask, table = node.process(field, method="absolute", threshold=0.5, direction="above")
|
||||
assert mask.dtype == np.uint8
|
||||
assert mask.shape == (64, 64)
|
||||
assert np.all(mask[:, :32] == 0)
|
||||
|
||||
Reference in New Issue
Block a user