@@ -0,0 +1,1140 @@
+%22%22%22Test dataset module.%0A%0AAuthor: Yuhuang Hu%0AEmail : yuhuang.hu@uzh.ch%0A%22%22%22%0A%0Aimport numpy as np%0Afrom simretina import dataset%0A%0Afrom nose.tools import assert_equal, assert_not_equal%0A%0A%0Adef test_get_lenna():%0A    %22%22%22Test dataset.get_lenna function.%22%22%22%0A    lenna, size = dataset.get_lenna()%0A%0A    assert_equal(lenna.shape, size)%0A    assert_not_equal(lenna, np.array(%5B%5D))%0A%0A%0Adef test_get_dog():%0A    %22%22%22Test dataset.get_dog function.%22%22%22%0A    dog, size = dataset.get_dog()%0A%0A    assert_equal(dog.shape, size)%0A    assert_not_equal(dog, np.array(%5B%5D))%0A%0A%0Adef test_get_yuhuang():%0A    %22%22%22Test dataset.get_yuhuang function.%22%22%22%0A    yuhuang, size = dataset.get_yuhuang()%0A%0A    assert_equal(yuhuang.shape, size)%0A    assert_not_equal(yuhuang, np.array(%5B%5D))%0A%0A%0Adef test_get_horse_riding():%0A    %22%22%22Test dataset.get_horse_riding function.%22%22%22%0A    horse_riding, size = dataset.get_horse_riding()%0A%0A    assert_equal(horse_riding%5B0%5D.shape, size)%0A    assert_not_equal(len(horse_riding), 0)%0A%0A%0Adef test_get_taichi():%0A    %22%22%22Test datset.get_taichi function.%22%22%22%0A    taichi, size = dataset.get_taichi()%0A%0A    assert_equal(taichi%5B0%5D.shape, size)%0A    assert_not_equal(len(taichi), 0)%0A

add tests for datasets