def test_with_ax(self):
fig, ax = plt.subplots()
fig1, ax1 = viz._check_ax_obj(ax)
nt.assert_true(isinstance(ax1, plt.Axes))
nt.assert_true(isinstance(fig1, plt.Figure))
nt.assert_true(ax1 is ax)
nt.assert_true(fig1 is fig)
def test_with_None(self):
fig1, ax1 = viz._check_ax_obj(None)
nt.assert_true(isinstance(ax1, plt.Axes))
nt.assert_true(isinstance(fig1, plt.Figure))
def test_bad_value(self):
viz._check_ax_obj('junk')
