Merge pull request #178 from brentp/ctable-strings

allow adding strings > len 1 to ctable

bcolz/ctable.py

@@ -371,7 +371,7 @@ def append(self, cols):
             if sclist and not hasattr(column, '__len__'):
                 clen2 = 1
             else:
-                clen2 = len(column)
+                clen2 = 1 if isinstance(column, _strtypes) else len(column)
             if clen >= 0 and clen != clen2:
                 raise ValueError(
                     "all cols in `cols` must have the same length")

bcolz/tests/test_ctable.py

@@ -2081,6 +2081,17 @@ def test_08(self):
 
         t.flush.assert_called_with()
 
+    def test_strings(self):
+        """Testing that we can add fixed length strings to a ctable"""
+        dtype = np.dtype([("a", "|S5"),
+                          ("b", np.uint8),
+                          ("c", np.int32),
+                          ("d", np.float32)])
+        t = bcolz.ctable(np.empty(0, dtype=dtype), mode="w")
+        t.append(("aaaaa", 23, 34567, 1.2355))
+        self.assertTrue(len(t) == 1)
+        self.assertTrue(t["a"][0] == b"aaaaa", t["a"][0])
+
 if __name__ == '__main__':
     unittest.main(verbosity=2)