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'
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): # 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')