improving test

Author: mmatera

test/test_custom_boxconstruct.py

@@ -4,6 +4,7 @@
 
 from mathics.core.expression import (Symbol, Atom, String, Expression)
 from mathics.builtin.base import BoxConstruct, AtomBuiltin, Predefined
+from mathics.builtin.graphics import GRAPHICS_OPTIONS
 
 class CustomBoxConstruct(BoxConstruct):
     def __init__(self, evaluation):
@@ -36,6 +37,39 @@ def apply_to_boxes(self, evaluation):
         'System`MakeBoxes[System`CustomAtom, StandardForm|TraditionalForm|OutputForm|InputForm]'
         return CustomBoxConstruct(evaluation=evaluation)
 
+
+class CustomGraphicsBox(BoxConstruct):
+    """
+
+    """
+    options = GRAPHICS_OPTIONS
+    attributes = ("HoldAll", "ReadProtected")
+
+
+    def apply_box(self, elems, evaluation, options):
+        """System`MakeBoxes[System`Graphics[elems_, System`OptionsPattern[System`Graphics]],
+        System`StandardForm|System`TraditionalForm|System`OutputForm]"""
+        instance = CustomGraphicsBox(*(elems._leaves), evaluation=evaluation)
+        return instance
+
+    def boxes_to_text(self, leaves=None, **options):
+        if leaves:
+           self._leaves = leaves 
+        return "--custom graphics--: I should plot " + self._leaves.__str__() +" items"
+
+    def boxes_to_tex(self, leaves=None, **options):
+        return "--custom graphics--: I should plot " + self._leaves.__str__() +" items" 
+
+    def boxes_to_xml(self, leaves=None, **options):
+        return "--custom graphics--: I should plot " + self._leaves.__str__() +" items"
+
+    def boxes_to_svg(self, evaluation):
+        return "--custom graphics--: I should plot " + self._leaves.__str__() +" items"
+
+    
+
+
+
 def test_custom_boxconstruct():
     defs = session.evaluation.definitions
     instance_custom_atom = CustomAtom(expression=False)
@@ -44,3 +78,12 @@ def test_custom_boxconstruct():
     formatted = session.format_result().boxes_to_xml()
     assert formatted == "CustomBoxConstruct<<[1, 2, 3]>>"
 
+
+
+def test_custom_graphicsbox_constructor():
+    defs = session.evaluation.definitions
+    instance_customgb_atom = CustomGraphicsBox(expression=False, evaluation=session.evaluation)
+    instance_customgb_atom.contribute(defs, is_pymodule=True)
+    expr = session.evaluate("MakeBoxes[Graphics[{Circle[{0,0},1]}], OutputForm]")
+    formatted = session.format_result().boxes_to_xml()
+    assert formatted == "--custom graphics--: I should plot (<Expression: System`Circle[System`List[0, 0], 1]>,) items"