Don't present integer k/v items as floats (#37)

v3io · Jun 10, 2020 · cebd7d9 · cebd7d9
1 parent 140bfba
commit cebd7d9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/tests/test_client.py b/tests/test_client.py
@@ -334,6 +334,7 @@ def test_emd_values(self):
         item = {
             item_key: {
                 'age': 42,
+                'pi': 3.14,
                 'feature': 'mustache',
                 'male': True,
                 'happy': False,
@@ -350,6 +351,9 @@ def test_emd_values(self):
 
         self.assertEqual(item[item_key], response.output.item)
 
+        for key, value in item[item_key].items():
+            self.assertEqual(type(value), type(response.output.item[key]))
+
     def test_emd(self):
         items = {
             'bob': {'age': 42, 'feature': 'mustache'},

diff --git a/v3io/dataplane/output.py b/v3io/dataplane/output.py
@@ -11,9 +11,14 @@ def _decode_typed_attributes(self, typed_attributes):
         for attribute_key, typed_attribute_value in future.utils.viewitems(typed_attributes):
             for attribute_type, attribute_value in future.utils.viewitems(typed_attribute_value):
                 if attribute_type == 'N':
-                    decoded_attribute = float(attribute_value)
+                    try:
+                        decoded_attribute = int(attribute_value)
+                    except ValueError:
+                        decoded_attribute = float(attribute_value)
                 elif attribute_type == 'B':
                     decoded_attribute = base64.b64decode(attribute_value)
+                elif attribute_type == 'S':
+                    decoded_attribute = str(attribute_value)
                 else:
                     decoded_attribute = attribute_value