def test_generate_axis_limit_commands_returns_empty_list_if_limits_not_changed(self): fig = plt.figure() ax = fig.add_subplot(111) ax.plot([-10, 10], [1, 2]) self.assertEqual([], generate_axis_limit_commands(ax)) plt.close() del fig
def get_axis_limit_cmds(ax, ax_object_var): """Get commands such as axes.set_xlim and axes.set_ylim""" axis_limit_cmds = generate_axis_limit_commands(ax) return [ "{ax_obj}.{cmd}".format(ax_obj=ax_object_var, cmd=cmd) for cmd in axis_limit_cmds ]
def test_generate_axis_limit_commands_returns_x_limit_command_if_x_limit_changed(self): fig = plt.figure() ax = fig.add_subplot(111) ax.plot([-10, 10], [1, 2]) ax.set_xlim([-5, 5]) self.assertEqual(['set_xlim([-5.0, 5.0])'], generate_axis_limit_commands(ax)) plt.close() del fig