51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
import numpy as np
|
|
import pytest
|
|
from tests.node_tests._shared import make_field
|
|
|
|
|
|
def test_canny_edge_detection():
|
|
from backend.nodes.feature_detection import FeatureDetection
|
|
|
|
node = FeatureDetection()
|
|
# Create a field with a sharp edge
|
|
data = np.zeros((64, 64))
|
|
data[:, 32:] = 1.0
|
|
field = make_field(data=data)
|
|
result, records = node.process(field, "canny", 1.0)
|
|
assert result.data.shape == (64, 64)
|
|
# Should detect some edge pixels
|
|
assert result.data.sum() > 0
|
|
assert isinstance(records, list)
|
|
|
|
|
|
def test_harris_corner_detection():
|
|
from backend.nodes.feature_detection import FeatureDetection
|
|
|
|
node = FeatureDetection()
|
|
# Create a field with a sharp corner (L-shape)
|
|
data = np.zeros((64, 64))
|
|
data[16:48, 16:48] = 1.0
|
|
field = make_field(data=data)
|
|
result, records = node.process(field, "harris", 1.0)
|
|
assert result.data.shape == (64, 64)
|
|
assert isinstance(records, list)
|
|
|
|
|
|
def test_canny_flat_field():
|
|
from backend.nodes.feature_detection import FeatureDetection
|
|
|
|
node = FeatureDetection()
|
|
field = make_field(data=np.ones((32, 32)))
|
|
result, records = node.process(field, "canny", 1.0)
|
|
# Flat field should have no edges
|
|
assert result.data.sum() == 0
|
|
|
|
|
|
def test_unknown_method():
|
|
from backend.nodes.feature_detection import FeatureDetection
|
|
|
|
node = FeatureDetection()
|
|
field = make_field(shape=(32, 32))
|
|
with pytest.raises(ValueError):
|
|
node.process(field, "unknown", 1.0)
|