Jdev

tests/test_convert_image_export.py

@@ -0,0 +1,52 @@
+import unittest
+import test_settings
+import ndio.remote.neurodata as neurodata
+import ndio.ramon
+import ndio.convert.png as ndpng
+import ndio.convert.tiff as ndtiff
+import numpy as np
+import os
+
+
+class TestDownload(unittest.TestCase):
+
+    def setUp(self):
+        self.hostname = test_settings.HOSTNAME
+        self.token_user = test_settings.NEURODATA
+        self.nd = neurodata(user_token=self.token_user, hostname=self.hostname)
+        self.image_data = np.zeros((512,512), dtype=np.uint8)
+
+        # self.image_data = (255.0 / data.max() * (data - data.min())).astype(np.uint8)
+
+    def test_export_load_png(self):
+
+        # if returns string, successful export
+        self.assertEqual(
+                ndpng.save("download.png", self.image_data),
+                "download.png")
+
+        # now confirm import works too
+        self.assertEqual(ndpng.load("download.png")[0][0],
+                         self.image_data[0][0])
+        self.assertEqual(ndpng.load("download.png")[10][10],
+                         self.image_data[10][10])
+        os.system("rm ./download.png")
+
+    def test_export_load_tiff(self):
+        # if returns string, successful export
+        self.assertEqual(
+                         ndtiff.save("download-1.tiff",
+                                     self.image_data),
+                         "download-1.tiff")
+
+        # now confirm import works too
+        self.assertEqual(
+                         ndtiff.load("download-1.tiff")[0][0],
+                         self.image_data[0][0])
+        self.assertEqual(
+                         ndtiff.load("download-1.tiff")[10][10],
+                         self.image_data[10][10])
+       	os.system("rm ./download-1.tiff")
+
+if __name__ == '__main__':
+    unittest.main()

tests/test_data.py

@@ -0,0 +1,50 @@
+import unittest
+import ndio.remote.neurodata as neurodata
+import ndio.remote.errors
+import numpy
+import h5py
+import os
+import test_settings
+
+HOSTNAME = test_settings.HOSTNAME
+NEURODATA = test_settings.NEURODATA
+TEST = test_settings.TEST
+
+class TestData(unittest.TestCase):
+
+    def setUp(self):
+        self.token_user = NEURODATA
+        hostname = HOSTNAME
+        self.nd = neurodata(self.token_user,
+                            hostname=hostname,
+                            check_tokens=True)
+        self.ramon_id = 1
+        dataset_name = 'demo1'
+        project_name = 'ndio_demos'
+        self.t = 'test_token'
+        self.c = 'test_channel'
+        self.nd.create_dataset(dataset_name, 100, 100, 100, 1.0,
+                                      1.0,
+                                      1.0)
+        self.nd.create_project(project_name, dataset_name,
+                                      'localhost',
+                                      1, 1, 'localhost', 'Redis')
+        self.nd.create_token(self.t, project_name, dataset_name, 1)
+        try:
+            self.nd.create_channel(self.c, project_name,
+                               dataset_name, 'timeseries',
+                               'uint8', 0, 500, propagate=1)
+        except:
+            pass
+
+        #random 3-d array
+        self.data = numpy.arange(30).reshape(2,3,5)
+
+    def test_post_cutout(self):
+        self.nd.post_cutout(self.t, self.c, 0, 0, 0, 0, self.data)
+
+
+
+
+if __name__ == '__main__':
+    unittest.main()