Fix issue where string_vector dies when given an int (#57)

Author: dan-blanchard

drmaa/helpers.py

@@ -238,7 +238,7 @@ def adapt_rusage(rusage):
     """
     rv = dict()
     for attr in attributes_iterator(rusage.contents):
-        
+
         k, v = attr.split('=',1)
         rv[k] = v
     return rv
@@ -307,7 +307,11 @@ def string_vector(v):
     vlen = len(v)
     values = (STRING * (vlen + 1))()
     for i, el in enumerate(v):
-        values[i] = STRING(el.encode(ENCODING) if isinstance(el, str) else el)
+        if isinstance(el, str):
+            el = el.encode(ENCODING)
+        elif not isinstance(el, bytes):
+            el = str(el).encode(ENCODING)
+        values[i] = STRING(el)
     values[vlen] = STRING()
     return values
 

test/testwrap.py

@@ -152,9 +152,9 @@ def xtest_tmp(self):
 
     def test_vector_attributes(self):
         """vector attributes work"""
-        args = ['10', 'de', 'arglebargle']
+        args = [10, 'de', 'arglebargle']
         self.jt.args = args
-        eq_(self.jt.args, args)
+        eq_(self.jt.args, ['10', 'de', 'arglebargle'])
         em = ['[email protected]', '[email protected]']
         self.jt.email = em
         eq_(self.jt.email, em)