mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-30 05:25:40 +08:00
Change test class to test functions
This commit is contained in:
Tony S Yu
+107
-102
@@ -1,123 +1,128 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
import os
|
||||
import tempfile
|
||||
|
||||
"""Tests for novice submodule.
|
||||
:license: modified BSD
|
||||
"""
|
||||
from numpy.testing import assert_equal, assert_raises
|
||||
from skimage import novice
|
||||
from skimage import data_dir
|
||||
|
||||
import os, tempfile
|
||||
from numpy.testing import TestCase, assert_equal, assert_raises
|
||||
from skimage import novice, data_dir
|
||||
|
||||
class TestNovice(TestCase):
|
||||
sample_path = os.path.join(data_dir, "elephant.png")
|
||||
small_sample_path = os.path.join(data_dir, "block.png")
|
||||
IMAGE_PATH = os.path.join(data_dir, "elephant.png")
|
||||
SMALL_IMAGE_PATH = os.path.join(data_dir, "block.png")
|
||||
|
||||
|
||||
def test_pic_info():
|
||||
pic = novice.open(IMAGE_PATH)
|
||||
assert_equal(pic.format, "png")
|
||||
assert_equal(pic.path, os.path.abspath(IMAGE_PATH))
|
||||
assert_equal(pic.size, (665, 500))
|
||||
assert_equal(pic.width, 665)
|
||||
assert_equal(pic.height, 500)
|
||||
assert_equal(pic.modified, False)
|
||||
assert_equal(pic.inflation, 1)
|
||||
|
||||
num_pixels = sum(1 for p in pic)
|
||||
assert_equal(num_pixels, pic.width * pic.height)
|
||||
|
||||
|
||||
def test_modify():
|
||||
pic = novice.open(SMALL_IMAGE_PATH)
|
||||
assert_equal(pic.modified, False)
|
||||
|
||||
for p in pic:
|
||||
if p.x < (pic.width / 2):
|
||||
p.red /= 2
|
||||
p.green /= 2
|
||||
p.blue /= 2
|
||||
|
||||
for p in pic:
|
||||
if p.x < (pic.width / 2):
|
||||
assert_equal(p.red <= 128, True)
|
||||
assert_equal(p.green <= 128, True)
|
||||
assert_equal(p.blue <= 128, True)
|
||||
|
||||
s = pic.size
|
||||
pic.size = (pic.width / 2, pic.height / 2)
|
||||
assert_equal(pic.size, (int(s[0] / 2), int(s[1] / 2)))
|
||||
|
||||
assert_equal(pic.modified, True)
|
||||
assert_equal(pic.path, None)
|
||||
|
||||
with tempfile.NamedTemporaryFile(suffix=".jpg") as tmp:
|
||||
pic.save(tmp.name)
|
||||
|
||||
def test_pic_info(self):
|
||||
pic = novice.open(self.sample_path)
|
||||
assert_equal(pic.format, "png")
|
||||
assert_equal(pic.path, os.path.abspath(self.sample_path))
|
||||
assert_equal(pic.size, (665, 500))
|
||||
assert_equal(pic.width, 665)
|
||||
assert_equal(pic.height, 500)
|
||||
assert_equal(pic.modified, False)
|
||||
assert_equal(pic.inflation, 1)
|
||||
assert_equal(pic.path, os.path.abspath(tmp.name))
|
||||
assert_equal(pic.format, "jpeg")
|
||||
|
||||
num_pixels = sum(1 for p in pic)
|
||||
assert_equal(num_pixels, pic.width * pic.height)
|
||||
|
||||
def test_modify(self):
|
||||
pic = novice.open(self.small_sample_path)
|
||||
assert_equal(pic.modified, False)
|
||||
def test_indexing():
|
||||
pic = novice.open(SMALL_IMAGE_PATH)
|
||||
|
||||
for p in pic:
|
||||
if p.x < (pic.width / 2):
|
||||
p.red /= 2
|
||||
p.green /= 2
|
||||
p.blue /= 2
|
||||
# Slicing
|
||||
pic[0:5, 0:5] = (0, 0, 0)
|
||||
for p in pic:
|
||||
if (p.x < 5) and (p.y < 5):
|
||||
assert_equal(p.rgb, (0, 0, 0))
|
||||
assert_equal(p.red, 0)
|
||||
assert_equal(p.green, 0)
|
||||
assert_equal(p.blue, 0)
|
||||
|
||||
for p in pic:
|
||||
if p.x < (pic.width / 2):
|
||||
assert_equal(p.red <= 128, True)
|
||||
assert_equal(p.green <= 128, True)
|
||||
assert_equal(p.blue <= 128, True)
|
||||
pic[:5, :5] = (255, 255, 255)
|
||||
for p in pic:
|
||||
if (p.x < 5) and (p.y < 5):
|
||||
assert_equal(p.rgb, (255, 255, 255))
|
||||
assert_equal(p.red, 255)
|
||||
assert_equal(p.green, 255)
|
||||
assert_equal(p.blue, 255)
|
||||
|
||||
s = pic.size
|
||||
pic.size = (pic.width / 2, pic.height / 2)
|
||||
assert_equal(pic.size, (int(s[0] / 2), int(s[1] / 2)))
|
||||
pic[5:pic.width, 5:pic.height] = (255, 0, 255)
|
||||
for p in pic:
|
||||
if (p.x >= 5) and (p.y >= 5):
|
||||
assert_equal(p.rgb, (255, 0, 255))
|
||||
assert_equal(p.red, 255)
|
||||
assert_equal(p.green, 0)
|
||||
assert_equal(p.blue, 255)
|
||||
|
||||
assert_equal(pic.modified, True)
|
||||
assert_equal(pic.path, None)
|
||||
pic[5:, 5:] = (0, 0, 255)
|
||||
for p in pic:
|
||||
if (p.x >= 5) and (p.y >= 5):
|
||||
assert_equal(p.rgb, (0, 0, 255))
|
||||
assert_equal(p.red, 0)
|
||||
assert_equal(p.green, 0)
|
||||
assert_equal(p.blue, 255)
|
||||
|
||||
with tempfile.NamedTemporaryFile(suffix=".jpg") as tmp:
|
||||
pic.save(tmp.name)
|
||||
# Outside bounds
|
||||
assert_raises(IndexError, lambda: pic[pic.width, pic.height])
|
||||
|
||||
assert_equal(pic.modified, False)
|
||||
assert_equal(pic.path, os.path.abspath(tmp.name))
|
||||
assert_equal(pic.format, "jpeg")
|
||||
# Negative indexing not supported
|
||||
assert_raises(IndexError, lambda: pic[-1, -1])
|
||||
assert_raises(IndexError, lambda: pic[-1:, -1:])
|
||||
|
||||
def test_indexing(self):
|
||||
pic = novice.open(self.small_sample_path)
|
||||
# Step sizes > 1 not supported
|
||||
assert_raises(IndexError, lambda: pic[::2, ::2])
|
||||
|
||||
# Slicing
|
||||
pic[0:5, 0:5] = (0, 0, 0)
|
||||
for p in pic:
|
||||
if (p.x < 5) and (p.y < 5):
|
||||
assert_equal(p.rgb, (0, 0, 0))
|
||||
assert_equal(p.red, 0)
|
||||
assert_equal(p.green, 0)
|
||||
assert_equal(p.blue, 0)
|
||||
|
||||
pic[:5, :5] = (255, 255, 255)
|
||||
for p in pic:
|
||||
if (p.x < 5) and (p.y < 5):
|
||||
assert_equal(p.rgb, (255, 255, 255))
|
||||
assert_equal(p.red, 255)
|
||||
assert_equal(p.green, 255)
|
||||
assert_equal(p.blue, 255)
|
||||
def test_slicing():
|
||||
cut = 40
|
||||
pic = novice.open(IMAGE_PATH)
|
||||
rest = pic.width - cut
|
||||
temp = pic[:cut, :]
|
||||
pic[:rest, :] = pic[cut:, :]
|
||||
pic[rest:, :] = temp
|
||||
|
||||
pic[5:pic.width, 5:pic.height] = (255, 0, 255)
|
||||
for p in pic:
|
||||
if (p.x >= 5) and (p.y >= 5):
|
||||
assert_equal(p.rgb, (255, 0, 255))
|
||||
assert_equal(p.red, 255)
|
||||
assert_equal(p.green, 0)
|
||||
assert_equal(p.blue, 255)
|
||||
pic_orig = novice.open(IMAGE_PATH)
|
||||
|
||||
pic[5:, 5:] = (0, 0, 255)
|
||||
for p in pic:
|
||||
if (p.x >= 5) and (p.y >= 5):
|
||||
assert_equal(p.rgb, (0, 0, 255))
|
||||
assert_equal(p.red, 0)
|
||||
assert_equal(p.green, 0)
|
||||
assert_equal(p.blue, 255)
|
||||
# Check center line
|
||||
half_height = int(pic.height/2)
|
||||
for p1 in pic_orig[rest:, half_height]:
|
||||
for p2 in pic[:cut, half_height]:
|
||||
assert p1.rgb == p2.rgb
|
||||
|
||||
# Outside bounds
|
||||
assert_raises(IndexError, lambda: pic[pic.width, pic.height])
|
||||
for p1 in pic_orig[:cut, half_height]:
|
||||
for p2 in pic[rest:, half_height]:
|
||||
assert p1.rgb == p2.rgb
|
||||
|
||||
# Negative indexing not supported
|
||||
assert_raises(IndexError, lambda: pic[-1, -1])
|
||||
assert_raises(IndexError, lambda: pic[-1:, -1:])
|
||||
|
||||
# Step sizes > 1 not supported
|
||||
assert_raises(IndexError, lambda: pic[::2, ::2])
|
||||
|
||||
def test_slicing(self):
|
||||
cut = 40
|
||||
pic = novice.open(self.sample_path)
|
||||
rest = pic.width - cut
|
||||
temp = pic[:cut, :]
|
||||
pic[:rest, :] = pic[cut:, :]
|
||||
pic[rest:, :] = temp
|
||||
|
||||
pic_orig = novice.open(self.sample_path)
|
||||
|
||||
# Check center line
|
||||
half_height = int(pic.height/2)
|
||||
for p1 in pic_orig[rest:, half_height]:
|
||||
for p2 in pic[:cut, half_height]:
|
||||
assert p1.rgb == p2.rgb
|
||||
|
||||
for p1 in pic_orig[:cut, half_height]:
|
||||
for p2 in pic[rest:, half_height]:
|
||||
assert p1.rgb == p2.rgb
|
||||
if __name__ == '__main__':
|
||||
from numpy import testing
|
||||
testing.run_module_suite()
|
||||
|
||||
Reference in New Issue
Block a user