🗓 Mar 28, 2022 12:26:40 PM

🐛 fixes #19
🔍 added some float type handlers and tests

chepy/core.py

@@ -7,6 +7,7 @@
 from pathlib import Path
 import subprocess
 import sys
+import struct
 import webbrowser
 from configparser import ConfigParser
 from importlib.machinery import SourceFileLoader
@@ -502,6 +503,8 @@ def _convert_to_bytes(self) -> bytes:
             return str(self.state).encode()
         elif isinstance(self.state, bytearray):
             return bytes(self.state)
+        elif isinstance(self.state, float):
+            return bytearray(struct.pack("f", self.state))
         else:  # pragma: no cover
             # todo check more types here
             raise NotImplementedError
@@ -536,6 +539,8 @@ def _convert_to_str(self) -> str:
             return str(self.state)
         elif isinstance(self.state, bytearray):
             return bytearray(self.state).decode()
+        elif isinstance(self.state, float):
+            return format(self.state, "f")
         else:  # pragma: no cover
             # todo check more types here
             raise NotImplementedError

chepy/modules/aritmeticlogic.py

@@ -81,7 +81,8 @@ def multiply(self, n: int) -> AritmeticLogicT:
 
     @ChepyDecorators.call_stack
     def divide(self, n: int) -> AritmeticLogicT:
-        """Divide a number to the state
+        """Divide a number to the state. Chepy is not optimized for float math. 
+        Subsequent methods may fail. 
 
         Args:
             n (int): Number to divide with

tests/test_aritmeticlogic.py

@@ -16,6 +16,9 @@ def test_multiply():
 def test_divide():
     assert Chepy("0x40").divide(2).o == 32
 
+def test_divide_float():
+    assert Chepy("179").divide(178).to_hex().o == b'17b8803f'
+
 
 def test_power():
     assert Chepy("0x02").power(2).o == 4