Update drawPrimitive to remove quotation marks from non-str objects (#45

)

CHANGELOG.md

@@ -23,6 +23,7 @@ and adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
 -   Fixed a bug where box fill colours would cover box text, and changed the implementation of `hide` style option.
 -   Removed double quotes when rendering objects of type `None`.
+-   Removed double quotes when rendering objects that are not of type `str`.
 
 ### 📚 Documentation and demo website changes
 

memory-viz/src/memory_model.ts

@@ -223,10 +223,10 @@ export class MemoryModel {
         let display_text;
         if (type === "bool") {
             display_text = value ? "True" : "False";
-        } else if (type === "None") {
-            display_text = "None";
-        } else {
+        } else if (type === "str") {
             display_text = JSON.stringify(value);
+        } else {
+            display_text = String(value);
         }
 
         if (value !== null && value !== undefined) {

memory-viz/src/tests/draw.spec.tsx

@@ -749,4 +749,20 @@ describe("draw function", () => {
         const svg: String = m.serializeSVG();
         expect(svg).toMatchSnapshot();
     });
+
+    it("renders range object", () => {
+        const objects: Array<Object> = [
+            {
+                type: "range",
+                id: 42,
+                value: "range(1, 5)",
+            },
+        ];
+        const m: InstanceType<typeof MemoryModel> = draw(objects, true, {
+            width: 1300,
+            roughjs_config: { options: { seed: 12345 } },
+        });
+        const svg: String = m.serializeSVG();
+        expect(svg).toMatchSnapshot();
+    });
 });