cp to tests

Author: ethho

tests/test_fetch_same.py

@@ -0,0 +1,62 @@
+from nose.tools import assert_equal
+from . import PREFIX, CONN_INFO
+import numpy as np
+import datajoint as dj
+
+schema = dj.Schema(PREFIX + "_fetch_same", connection=dj.conn(**CONN_INFO))
+
+
+class TestFetchSame:
+    @classmethod
+    def setup_class(cls):
+        @schema
+        class ProjData(dj.Manual):
+            definition = """
+            id : int
+            ---
+            resp : float
+            sim  : float
+            big : longblob
+            blah : varchar(10)
+            """
+
+        ProjData().insert(
+            [
+                {"id": 0, "resp": 20.33, "sim": 45.324, "big": 3, "blah": "yes"},
+                {
+                    "id": 1,
+                    "resp": 94.3,
+                    "sim": 34.23,
+                    "big": {"key1": np.random.randn(20, 10)},
+                    "blah": "si",
+                },
+                {
+                    "id": 2,
+                    "resp": 1.90,
+                    "sim": 10.23,
+                    "big": np.random.randn(4, 2),
+                    "blah": "sim",
+                },
+            ]
+        )
+
+        cls.projdata = ProjData()
+
+    def test_object_conversion_one(self):
+        new = self.projdata.proj(sub="resp").fetch("sub")
+        assert_equal(new.dtype, np.float64)
+
+    def test_object_conversion_two(self):
+        [sub, add] = self.projdata.proj(sub="resp", add="sim").fetch("sub", "add")
+        assert_equal(sub.dtype, np.float64)
+        assert_equal(add.dtype, np.float64)
+
+    def test_object_conversion_all(self):
+        new = self.projdata.proj(sub="resp", add="sim").fetch()
+        assert_equal(new["sub"].dtype, np.float64)
+        assert_equal(new["add"].dtype, np.float64)
+
+    def test_object_no_convert(self):
+        new = self.projdata.fetch()
+        assert_equal(new["big"].dtype, "object")
+        assert_equal(new["blah"].dtype, "object")