Convert NumPy arrays when updating _SyncedDict. (#185)

* These tests should fail, demonstrating issue 184.

* Convert NumPy types when calling _dfs_convert.

* Update changelog.txt

changelog.txt

@@ -24,6 +24,7 @@ next
  - Fix: Passing an instance of dict to `H5Store.setdefault()` will return an instance of `H5Group` instead of a dict (#180).
  - Improve the representation (return value of `repr()`) of instances of `H5Group`.
  - Improve the representation (return value of `repr()`) of instances of `SyncedAttrDict`.
+ - Fix error with storing numpy arrays and scalars in a synced dictionary (e.g. `job.statepoint`, `job.document`) (#184).
 
 
 [1.0.0] -- 2019-02-28

signac/core/synceddict.py

@@ -15,6 +15,11 @@
 else:
     from collections.abc import Mapping
     from collections.abc import MutableMapping
+try:
+    import numpy
+    NUMPY = True
+except ImportError:
+    NUMPY = False
 
 
 logger = logging.getLogger(__name__)
@@ -107,6 +112,11 @@ def _dfs_convert(self, root):
             return ret
         elif type(root) is list:
             return _SyncedList(root, self)
+        elif NUMPY:
+            if isinstance(root, numpy.number):
+                return root.item()
+            elif isinstance(root, numpy.ndarray):
+                return _SyncedList(root.tolist(), self)
         return root
 
     @classmethod

tests/test_numpy_integration.py

@@ -5,6 +5,7 @@
 from test_project import BaseProjectTest
 try:
     import numpy    # noqa
+    import numpy.testing
     NUMPY = True
 except ImportError:
     NUMPY = False
@@ -19,6 +20,7 @@ def test_store_number_in_sp_and_doc(self):
             b = numpy.float64(i) if i % 2 else numpy.float32(i)
             job = self.project.open_job(dict(a=a))
             job.doc.b = b
+            numpy.testing.assert_equal(job.doc.b, b)
         for i, job in enumerate(sorted(self.project, key=lambda job: job.sp.a)):
             self.assertEqual(job.sp.a, i)
             self.assertEqual(job.doc.b, i)
@@ -34,6 +36,7 @@ def test_store_array_in_doc(self):
         for i in range(10):
             job = self.project.open_job(dict(a=i))
             job.doc.array = numpy.ones(3) * i
+            numpy.testing.assert_equal(job.doc.array, numpy.ones(3) * i)
         for i, job in enumerate(sorted(self.project, key=lambda job: job.sp.a)):
             self.assertEqual(i, job.sp.a)
             self.assertTrue(((numpy.array([i, i, i]) == job.doc.array).all()))