Fix datajoint#1103, fix test_blob (for some numpy versions)

Author: dimitri-yatsenko

datajoint/table.py

@@ -196,7 +196,6 @@ def parents(self, primary=None, as_objects=False, foreign_key_info=False):
 
     def children(self, primary=None, as_objects=False, foreign_key_info=False):
         """
-
         :param primary: if None, then all children are returned. If True, then only foreign keys composed of
             primary key attributes are considered.  If False, return foreign keys including at least one
             secondary attribute.
@@ -230,7 +229,6 @@ def descendants(self, as_objects=False):
 
     def ancestors(self, as_objects=False):
         """
-
         :param as_objects: False - a list of table names; True - a list of table objects.
         :return: list of tables ancestors in topological order.
         """
@@ -246,6 +244,7 @@ def parts(self, as_objects=False):
 
         :param as_objects: if False (default), the output is a dict describing the foreign keys. If True, return table objects.
         """
+        self.connection.dependencies.load(force=False)
         nodes = [
             node
             for node in self.connection.dependencies.nodes
@@ -427,7 +426,8 @@ def insert(
             self.connection.query(query)
             return
 
-        field_list = []  # collects the field list from first row (passed by reference)
+        # collects the field list from first row (passed by reference)
+        field_list = []
         rows = list(
             self.__make_row_to_insert(row, field_list, ignore_extra_fields)
             for row in rows
@@ -520,7 +520,8 @@ def cascade(table):
                     delete_count = table.delete_quick(get_count=True)
                 except IntegrityError as error:
                     match = foreign_key_error_regexp.match(error.args[0]).groupdict()
-                    if "`.`" not in match["child"]:  # if schema name missing, use table
+                    # if schema name missing, use table
+                    if "`.`" not in match["child"]:
                         match["child"] = "{}.{}".format(
                             table.full_table_name.split(".")[0], match["child"]
                         )
@@ -962,7 +963,8 @@ def lookup_class_name(name, context, depth=3):
     while nodes:
         node = nodes.pop(0)
         for member_name, member in node["context"].items():
-            if not member_name.startswith("_"):  # skip IPython's implicit variables
+            # skip IPython's implicit variables
+            if not member_name.startswith("_"):
                 if inspect.isclass(member) and issubclass(member, Table):
                     if member.full_table_name == name:  # found it!
                         return ".".join([node["context_name"], member_name]).lstrip(".")

tests/test_blob.py

@@ -124,16 +124,20 @@ def test_pack():
     assert x == unpack(pack(x)), "Set did not pack/unpack correctly"
 
     x = tuple(range(10))
-    assert x == unpack(pack(range(10))), "Iterator did not pack/unpack correctly"
+    assert x == unpack(
+        pack(range(10))), "Iterator did not pack/unpack correctly"
 
     x = Decimal("1.24")
-    assert x == approx(unpack(pack(x))), "Decimal object did not pack/unpack correctly"
+    assert x == approx(
+        unpack(pack(x))), "Decimal object did not pack/unpack correctly"
 
     x = datetime.now()
-    assert x == unpack(pack(x)), "Datetime object did not pack/unpack correctly"
+    assert x == unpack(
+        pack(x)), "Datetime object did not pack/unpack correctly"
 
     x = np.bool_(True)
-    assert x == unpack(pack(x)), "Numpy bool object did not pack/unpack correctly"
+    assert x == unpack(
+        pack(x)), "Numpy bool object did not pack/unpack correctly"
 
     x = "test"
     assert x == unpack(pack(x)), "String object did not pack/unpack correctly"
@@ -154,13 +158,15 @@ def test_recarrays():
     x = x.view(np.recarray)
     assert_array_equal(x, unpack(pack(x)))
 
-    x = np.array([(3, 4)], dtype=[("tmp0", float), ("tmp1", "O")]).view(np.recarray)
+    x = np.array([(3, 4)], dtype=[("tmp0", float),
+                 ("tmp1", "O")]).view(np.recarray)
     assert_array_equal(x, unpack(pack(x)))
 
 
 def test_object_arrays():
     x = np.array(((1, 2, 3), True), dtype="object")
-    assert_array_equal(x, unpack(pack(x)), "Object array did not serialize correctly")
+    assert_array_equal(x, unpack(pack(x)),
+                       "Object array did not serialize correctly")
 
 
 def test_complex():
@@ -170,10 +176,12 @@ def test_complex():
     z = np.random.randn(10) + 1j * np.random.randn(10)
     assert_array_equal(z, unpack(pack(z)), "Arrays do not match!")
 
-    x = np.float32(np.random.randn(3, 4, 5)) + 1j * np.float32(np.random.randn(3, 4, 5))
+    x = np.float32(np.random.randn(3, 4, 5)) + 1j * \
+        np.float32(np.random.randn(3, 4, 5))
     assert_array_equal(x, unpack(pack(x)), "Arrays do not match!")
 
-    x = np.int16(np.random.randn(1, 2, 3)) + 1j * np.int16(np.random.randn(1, 2, 3))
+    x = np.int16(np.random.randn(1, 2, 3)) + 1j * \
+        np.int16(np.random.randn(1, 2, 3))
     assert_array_equal(x, unpack(pack(x)), "Arrays do not match!")
 
 
@@ -185,7 +193,8 @@ def test_insert_longblob(schema_any):
 
     query_mym_blob = {"id": 1, "data": np.array([1, 2, 3])}
     Longblob.insert1(query_mym_blob)
-    assert (Longblob & "id=1").fetch1()["data"].all() == query_mym_blob["data"].all()
+    assert_array_equal(
+        (Longblob & "id=1").fetch1()["data"], query_mym_blob["data"])
     (Longblob & "id=1").delete()
 
 
@@ -214,11 +223,14 @@ def test_insert_longblob_32bit(schema_any, enable_feature_32bit_dims):
                     )
                 ]
             ],
-            dtype=[("hits", "O"), ("sides", "O"), ("tasks", "O"), ("stage", "O")],
+            dtype=[("hits", "O"), ("sides", "O"),
+                   ("tasks", "O"), ("stage", "O")],
         ),
     }
     assert fetched["id"] == expected["id"]
-    assert np.array_equal(fetched["data"], expected["data"])
+    for name in expected['data'][0][0].dtype.names:
+        assert_array_equal(
+            expected['data'][0][0][name], fetched['data'][0][0][name])
     (Longblob & "id=1").delete()