Fix string encoding

CourtBouillon · Aug 22, 2021 · 0f7c8e9 · 0f7c8e9
1 parent 960b012
commit 0f7c8e9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/pydyf/__init__.py b/pydyf/__init__.py
@@ -17,12 +17,12 @@ def _to_bytes(item):
         return item.data
     elif isinstance(item, float):
         if item.is_integer():
-            return f'{int(item):d}'.encode()
+            return f'{int(item):d}'.encode('ascii')
         else:
-            return f'{item:f}'.encode()
+            return f'{item:f}'.encode('ascii')
     elif isinstance(item, int):
-        return f'{item:d}'.encode()
-    return str(item).encode()
+        return f'{item:d}'.encode('ascii')
+    return str(item).encode('ascii')
 
 
 class Object:

diff --git a/tests/test_pydyf.py b/tests/test_pydyf.py
@@ -700,3 +700,9 @@ def test_text():
         __________
         __________
     ''')
+
+
+def test_string_encoding():
+    assert pydyf.String('abc').data == b'(abc)'
+    assert pydyf.String('déf').data == b'<feff006400e90066>'
+    assert pydyf.String('♡').data == b'<feff2661>'