def test_axes_without_shift(): """Test whether axes are not shifted when origin is in plot range.""" G = GraphScene( x_min=-1, x_max=2, x_axis_label="", y_min=-2, y_max=5, y_axis_label="" ) G.setup_axes() assert all(np.isclose(G.graph_origin, G.x_axis.n2p(0))) assert all(np.isclose(G.graph_origin, G.y_axis.n2p(0)))
def test_axes_with_y_shift(): """Test whether y-axis is shifted when 0 is not in plot range of y-axis.""" G = GraphScene( x_min=-1, x_max=2, x_axis_label="", y_min=1, y_max=5, y_axis_label="" ) G.setup_axes() assert all(np.isclose(G.graph_origin, G.x_axis.n2p(0))) assert all(np.isclose(G.graph_origin, G.y_axis.n2p(1)))
def test_axes_with_xy_shift(): """Test whether both axes are shifted when origin is not in plot range.""" G = GraphScene( x_min=1, x_max=5, x_axis_label="", y_min=-5, y_max=-1, y_axis_label="" ) G.setup_axes() assert all(np.isclose(G.graph_origin, G.x_axis.n2p(1))) assert all(np.isclose(G.graph_origin, G.y_axis.n2p(-5)))
def __init__(self, **kwargs): GraphScene.__init__(self, x_min=0, x_max=12, y_min=-2, y_max=2, graph_origin=np.array((-5.0, 0.0, 0.0)), axes_color=nq_textcolors["white"], x_labeled_nums=range(0, 12, 2), **kwargs)