add test for issue1440

Author: jswhit2

test/test_stringarr.py

@@ -3,6 +3,7 @@
 import unittest
 import os
 from numpy.testing import assert_array_equal, assert_array_almost_equal
+import numpy as np
 
 def generateString(length, alphabet=string.ascii_letters + string.digits + string.punctuation):
     return(''.join([random.choice(alphabet) for i in range(length)]))
@@ -20,6 +21,10 @@ def generateString(length, alphabet=string.ascii_letters + string.digits + strin
 datau = data.astype('U')
 datac = stringtochar(data, encoding='ascii')
 
+nx, n_strlen = 3, 10
+unicode_strings = np.array(['Münster', 'London', 'Amsterdam'],dtype='U'+str(n_strlen))
+unicode_strings2 = np.array(['Münster', 'Liége', 'Amsterdam'],dtype='U'+str(n_strlen))
+
 class StringArrayTestCase(unittest.TestCase):
 
     def setUp(self):
@@ -28,6 +33,8 @@ def setUp(self):
         nc.createDimension('n1',None)
         nc.createDimension('n2',n2)
         nc.createDimension('nchar',nchar)
+        nc.createDimension("x", nx)
+        nc.createDimension("nstr", n_strlen)
         v = nc.createVariable('strings','S1',('n1','n2','nchar'))
         v2 = nc.createVariable('strings2','S1',('n1','n2','nchar'))
         # if _Encoding set, string array should automatically be converted
@@ -44,6 +51,11 @@ def setUp(self):
         v2[-1,-1] = data[-1,-1].tobytes() # write single python string
         # _Encoding should be ignored if an array of characters is specified
         v3[:] = stringtochar(data, encoding='ascii')
+        # test unicode strings (issue #1440)
+        v4 = nc.createVariable("strings4", "S1", dimensions=("x", "nstr",))
+        v4._Encoding = "UTF-8"
+        v4[:] = unicode_strings
+        v4[1] = "Liége"
         nc.close()
 
     def tearDown(self):
@@ -57,6 +69,8 @@ def runTest(self):
         v = nc.variables['strings']
         v2 = nc.variables['strings2']
         v3 = nc.variables['strings3']
+        v4 = nc.variables['strings4']
+        assert np.all(v4[:]==unicode_strings2)
         assert v.dtype.str[1:] in ['S1','U1']
         assert v.shape == (nrecs,n2,nchar)
         for nrec in range(nrecs):