class PrettyPrintOptionsTest(CustomBackendTestCase):
def setUp(self):
self.current_backend = Store.current_backend
self.pprinter = PrettyPrinter(show_options=True)
self.register_custom(TestObj, 'backend_1', ['plot_custom1'], ['style_custom1'])
self.register_custom(Overlay, 'backend_1', ['plot_custom1'])
self.register_custom(Layout, 'backend_1', ['plot_custom1'])
self.register_custom(TestObj, 'backend_2', ['plot_custom2'])
Store.current_backend = 'backend_1'
def test_element_options(self):
element = TestObj(None).opts(style_opt1='A', backend='backend_1')
r = self.pprinter.pprint(element)
self.assertEqual(r, ":TestObj\n | Options(style_opt1='A')")
def test_element_options_wrapping(self):
element = TestObj(None).opts(plot_opt1='A'*40, style_opt1='B'*40, backend='backend_1')
r = self.pprinter.pprint(element)
self.assertEqual(r, ":TestObj\n | Options(plot_opt1='AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA',\n | style_opt1='BBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBB')")
def test_overlay_options(self):
overlay = (TestObj(None) * TestObj(None)).opts(plot_opt1='A')
r = self.pprinter.pprint(overlay)
self.assertEqual(r, ":Overlay\n | Options(plot_opt1='A')\n .Element.I :TestObj\n .Element.II :TestObj")
def test_overlay_nested_options(self):
overlay = (TestObj(None) * TestObj(None)).opts('TestObj', plot_opt1='A', style_opt1='A')
r = self.pprinter.pprint(overlay)
self.assertEqual(r, ":Overlay\n .Element.I :TestObj\n | Options(plot_opt1='A', style_opt1='A')\n .Element.II :TestObj\n | Options(plot_opt1='A', style_opt1='A')")
def test_layout_options(self):
overlay = (TestObj(None) + TestObj(None)).opts(plot_opt1='A')
r = self.pprinter.pprint(overlay)
self.assertEqual(r, ":Layout\n | Options(plot_opt1='A')\n .Element.I :TestObj\n .Element.II :TestObj")
def test_layout_nested_options(self):
overlay = (TestObj(None) + TestObj(None)).opts('TestObj', plot_opt1='A', style_opt1='A')
r = self.pprinter.pprint(overlay)
self.assertEqual(r, ":Layout\n .Element.I :TestObj\n | Options(plot_opt1='A', style_opt1='A')\n .Element.II :TestObj\n | Options(plot_opt1='A', style_opt1='A')")
def test_curve_pprint_repr(self):
# Ensure it isn't a bytes object with the 'b' prefix
expected = "':Curve [x] (y)'"
r = PrettyPrinter.pprint(Curve([1,2,3]))
self.assertEqual(repr(r), expected)
def test_overlay_repr1(self):
expected = ':Overlay\n .Value.Label :Element\n .Value.I :Element'
o = self.element1 * self.element2
r = PrettyPrinter.pprint(o)
self.assertEqual(r, expected)
def test_element_repr1(self):
r = PrettyPrinter.pprint(self.element1)
self.assertEqual(r, ':Element')
def test_curve_pprint_repr(self):
# Ensure it isn't a bytes object with the 'b' prefix
expected = "':Curve [x] (y)'"
r = PrettyPrinter.pprint(Curve([1, 2, 3]))
self.assertEqual(repr(r), expected)
def test_overlay_repr1(self):
expected = ':Overlay\n .Value.Label :Element\n .Value.I :Element'
o = self.element1 * self.element2
r = PrettyPrinter.pprint(o)
self.assertEqual(r, expected)
def test_element_repr1(self):
r = PrettyPrinter.pprint(self.element1)
self.assertEqual(r, ':Element')
def test_overlay_repr1(self):
# Pointless trailing space...
expected = 'Overlay.Value\n*--Element.Value.Label\n*--Element.Value '
o = self.element1 * self.element2
r = PrettyPrinter.pprint(o)
self.assertEqual(r, expected)
def test_element_repr2(self):
r = PrettyPrinter.pprint(self.element2)
self.assertEqual(r, 'Element.Value')
def test_element_repr1(self):
r = PrettyPrinter.pprint(self.element1)
self.assertEqual(r, 'Element.Value.Label')