Changeset 6415 for branches/numpy/anuga/utilities/test_system_tools.py

Timestamp:

Feb 25, 2009, 11:44:06 AM (15 years ago)

Author:

rwilson

Message:

Added routines to encode/decode arrays of strings (and test).

File:

• branches/numpy/anuga/utilities/test_system_tools.py (modified) (1 diff)

branches/numpy/anuga/utilities/test_system_tools.py

@@ -183 +183 @@
 
 ################################################################################
+# Test the string_to_char() and char_to_string() utility functions.
+################################################################################
+
+    def test_string_to_char(self):
+        import random
+
+        MAX_CHARS = 10
+        MAX_ENTRIES = 100000
+        A_INT = ord('a')
+        Z_INT = ord('z')
+
+        # generate some random strings in a list, with guaranteed lengths
+        str_list = ['x' * MAX_CHARS]        # make first maximum length
+        for entry in xrange(MAX_ENTRIES):
+            length = random.randint(1, MAX_CHARS)
+            s = ''
+            for c in range(length):
+                s += chr(random.randint(A_INT, Z_INT))
+            str_list.append(s)
+
+        x = string_to_char(str_list)
+        new_str_list = char_to_string(x)
+
+        self.failUnlessEqual(new_str_list, str_list)
+
+
+################################################################################
+# Test the raw I/O to NetCDF files of string data encoded/decoded with 
+# string_to_char() and char_to_string().
+################################################################################
+
+################################################################################
 
 if __name__ == "__main__":
-    suite = unittest.makeSuite(Test_system_tools, 'test')
+    #suite = unittest.makeSuite(Test_system_tools, 'test')
+    suite = unittest.makeSuite(Test_system_tools, 'test_string_to_char')
     runner = unittest.TextTestRunner()
     runner.run(suite)