handle single-value labels and adding tests

Author: sprenger

nixio/data_array.py

@@ -91,7 +91,11 @@ def append_set_dimension(self, labels=None):
         index = len(self.dimensions) + 1
         setdim = SetDimension.create_new(self, index)
         if labels is not None:
-            if not isinstance(labels, (list, str)):
+            if not hasattr(labels, '__iter__'):
+                labels = str(labels)
+            if isinstance(labels, str):
+                labels = [labels]
+            if not isinstance(labels, list):
                 labels = list(labels)
             setdim.labels = labels
         if self.file.auto_update_timestamps:

nixio/test/test_dimensions.py

@@ -153,11 +153,21 @@ def test_set_dim_label_resize(self):
         setdim.labels = newlabels
         assert tuple(newlabels) == setdim.labels
 
-    def test_set_dim_label_array(self):
+    def test_set_dim_labels_array(self):
         labels = np.array(["A", "B"])
         setdim = self.array.append_set_dimension(labels)
         assert tuple(labels) == setdim.labels
 
+    def test_set_dim_labels_single_string(self):
+        labels = 'Sample 1'
+        setdim = self.array.append_set_dimension(labels)
+        assert tuple([labels]) == setdim.labels
+
+    def test_set_dim_labels_single_float(self):
+        labels = 1000
+        setdim = self.array.append_set_dimension(labels)
+        assert tuple([str(labels)]) == setdim.labels
+
     def test_range_dim_ticks_resize(self):
         rangedim = self.array.append_range_dimension([1, 2, 100])
         ticks = [1, 1, 30]