def test_disable_matplotlib_gui(self):
str_inp = "\nimport matplotlib as mp\nmp.use('PS')\n"
str_exp = ("\nimport matplotlib as _mpl;_mpl.use('Agg');"
"import matplotlib as mp\n\n")
str_out = iw.disable_matplotlib_gui(str_inp)
self.assertEqual(str_out, str_exp)
str_inp = "\nimport matplotlib.pyplot as plt\nplt.ion()\n"
str_exp = ("\nimport matplotlib as _mpl;_mpl.use('Agg');"
"import matplotlib.pyplot as plt\nplt.ioff()\n")
str_out = iw.disable_matplotlib_gui(str_inp)
self.assertEqual(str_out, str_exp)
def test_disable_matplotlib_gui(self):
str_inp = "if 1:\n\timport matplotlib as mp\nmp.use('PS')\n"
str_exp = ("if 1:\n\timport matplotlib as _mpl;_mpl.use('Agg');"
"import matplotlib as mp\n#mp.use('PS')\n")
str_out = iw.disable_matplotlib_gui(str_inp)
self.assertEqual(str_out, str_exp)
str_inp = "if 1:\n import matplotlib.pyplot as plt\nplt.ion()\n"
str_exp = ("if 1:\n import matplotlib as _mpl;_mpl.use('Agg');"
"import matplotlib.pyplot as plt\n#plt.ion()\n")
str_out = iw.disable_matplotlib_gui(str_inp)
self.assertEqual(str_out, str_exp)
str_inp = "if 1:\n\tget_ipython(\n).run_line_magic('matplotlib', 'x')\n"
str_exp = "if 1:\n#\tget_ipython(\n#).run_line_magic('matplotlib', 'x')\n"
str_out = iw.disable_matplotlib_gui(str_inp)
self.assertEqual(str_out, str_exp)