mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-20 12:40:31 +08:00
Changed variable names for readbility, fixed PEP8 compliance
This commit is contained in:
@@ -65,7 +65,7 @@ def test_color_3d():
|
||||
|
||||
assert_equal(len(np.unique(seg)), 8)
|
||||
for s, c in zip(slices, range(8)):
|
||||
assert_equal(seg[s], c+1)
|
||||
assert_equal(seg[s], c + 1)
|
||||
|
||||
|
||||
def test_gray_3d():
|
||||
@@ -76,7 +76,7 @@ def test_gray_3d():
|
||||
midpoint = dim_size // 2
|
||||
slices.append((slice(None, midpoint), slice(midpoint, None)))
|
||||
slices = list(it.product(*slices))
|
||||
shades = np.arange(0, 1.000001, 1.0/7)
|
||||
shades = np.arange(0, 1.000001, 1.0 / 7)
|
||||
for s, sh in zip(slices, shades):
|
||||
img[s] = sh
|
||||
img += 0.001 * rnd.normal(size=img.shape)
|
||||
@@ -87,7 +87,7 @@ def test_gray_3d():
|
||||
|
||||
assert_equal(len(np.unique(seg)), 8)
|
||||
for s, c in zip(slices, range(8)):
|
||||
assert_equal(seg[s], c+1)
|
||||
assert_equal(seg[s], c + 1)
|
||||
|
||||
|
||||
def test_list_sigma():
|
||||
@@ -106,7 +106,7 @@ def test_spacing():
|
||||
img = np.array([[1, 1, 1, 0, 0],
|
||||
[1, 1, 0, 0, 0]], np.float)
|
||||
result_non_spaced = np.array([[0, 0, 0, 1, 1],
|
||||
[0, 0, 1, 1, 1]], np.int) +1
|
||||
[0, 0, 1, 1, 1]], np.int) + 1
|
||||
result_spaced = np.array([[0, 0, 0, 0, 0],
|
||||
[1, 1, 1, 1, 1]], np.int) + 1
|
||||
img += 0.1 * rnd.normal(size=img.shape)
|
||||
@@ -124,7 +124,30 @@ def test_invalid_lab_conversion():
|
||||
assert_raises(ValueError, slic, img, multichannel=True, convert2lab=True)
|
||||
|
||||
|
||||
def test_enforce_connectivity():
|
||||
img = np.array([[0, 0, 0, 1, 1, 1],
|
||||
[1, 0, 0, 1, 1, 0],
|
||||
[0, 0, 0, 1, 1, 0]], np.float)
|
||||
|
||||
segments_connected = slic(img, 2, compactness=0.0001,
|
||||
enforce_connectivity=True,
|
||||
convert2lab=False)
|
||||
segments_disconnected = slic(img, 2, compactness=0.0001,
|
||||
enforce_connectivity=False,
|
||||
convert2lab=False)
|
||||
|
||||
result_connected = np.array([[1, 1, 1, 2, 2, 2],
|
||||
[1, 1, 1, 2, 2, 2],
|
||||
[1, 1, 1, 2, 2, 2]], np.float)
|
||||
|
||||
result_disconnected = np.array([[1, 1, 1, 2, 2, 2],
|
||||
[2, 1, 1, 2, 2, 1],
|
||||
[1, 1, 1, 2, 2, 1]], np.float)
|
||||
|
||||
assert_equal(segments_connected, result_connected)
|
||||
assert_equal(segments_disconnected, result_disconnected)
|
||||
|
||||
if __name__ == '__main__':
|
||||
from numpy import testing
|
||||
|
||||
testing.run_module_suite()
|
||||
|
||||
Reference in New Issue
Block a user