def test_matplotlib_colors(): """Examples from the 'colors' notebook.""" try: name = 'test' tmp_file = TmpFileManager.tmp_file p = plot(sin(x)) p[0].line_color = lambda a: a p.save(tmp_file('%s_colors_line_arity1' % name)) p[0].line_color = lambda a, b: b p.save(tmp_file('%s_colors_line_arity2' % name)) p = plot(x * sin(x), x * cos(x), (x, 0, 10)) p[0].line_color = lambda a: a p.save(tmp_file('%s_colors_param_line_arity1' % name)) p[0].line_color = lambda a, b: a p.save(tmp_file('%s_colors_param_line_arity2a' % name)) p[0].line_color = lambda a, b: b p.save(tmp_file('%s_colors_param_line_arity2b' % name)) del p p = plot3d_parametric_line( sin(x) + 0.1 * sin(x) * cos(7 * x), cos(x) + 0.1 * cos(x) * cos(7 * x), 0.1 * sin(7 * x), (x, 0, 2 * pi)) p[0].line_color = lambda a: sin(4 * a) p.save(tmp_file('%s_colors_3d_line_arity1' % name)) p[0].line_color = lambda a, b: b p.save(tmp_file('%s_colors_3d_line_arity2' % name)) p[0].line_color = lambda a, b, c: c p.save(tmp_file('%s_colors_3d_line_arity3' % name)) del p p = plot3d(sin(x) * y, (x, 0, 6 * pi), (y, -5, 5)) p[0].surface_color = lambda a: a p.save(tmp_file('%s_colors_surface_arity1' % name)) p[0].surface_color = lambda a, b: b p.save(tmp_file('%s_colors_surface_arity2' % name)) p[0].surface_color = lambda a, b, c: c p.save(tmp_file('%s_colors_surface_arity3a' % name)) p[0].surface_color = lambda a, b, c: sqrt((a - 3 * pi)**2 + b**2) p.save(tmp_file('%s_colors_surface_arity3b' % name)) del p p = plot3d_parametric_surface(x * cos(4 * y), x * sin(4 * y), y, (x, -1, 1), (y, -1, 1)) p[0].surface_color = lambda a: a p.save(tmp_file('%s_colors_param_surf_arity1' % name)) p[0].surface_color = lambda a, b: a * b p.save(tmp_file('%s_colors_param_surf_arity2' % name)) p[0].surface_color = lambda a, b, c: sqrt(a**2 + b**2 + c**2) p.save(tmp_file('%s_colors_param_surf_arity3' % name)) del p finally: TmpFileManager.cleanup()
def test_matplotlib_intro(): """Examples from the 'introduction' notebook.""" try: name = 'test' tmp_file = TmpFileManager.tmp_file p = plot(x, adaptive=False) assert str(p) == """Plot object containing: [0]: cartesian line: x for x over (-10.0, 10.0)""" p = plot(x * sin(x), x * cos(x)) p.extend(p) p[0].line_color = lambda a: a p[1].line_color = 'b' p.title = 'Big title' p.xlabel = 'the x axis' p[1].label = 'straight line' p.legend = True p.aspect_ratio = (1, 1) p.xlim = (-15, 20) p.save(tmp_file('%s_basic_options_and_colors' % name)) p.extend(plot(x + 1)) p.append(plot(x + 3, x**2)[1]) p.save(tmp_file('%s_plot_extend_append' % name)) p[2] = plot(x**2, (x, -2, 3)) p.save(tmp_file('%s_plot_setitem' % name)) del p p = plot(sin(x), (x, -2 * pi, 4 * pi)) p.save(tmp_file('%s_line_explicit' % name)) del p p = plot(sin(x)) p.save(tmp_file('%s_line_default_range' % name)) del p p = plot((x**2, (x, -5, 5)), (x**3, (x, -3, 3))) p.save(tmp_file('%s_line_multiple_range' % name)) del p pytest.raises(ValueError, lambda: plot(x, y)) # parametric 2d plots. # Single plot with default range. plot_parametric(sin(x), cos(x)).save(tmp_file()) # Single plot with range. p = plot_parametric(sin(x), cos(x), (x, -5, 5)) assert str(p) == """Plot object containing: [0]: parametric cartesian line: (sin(x), cos(x)) for x over (-5.0, 5.0)""" p.save(tmp_file('%s_parametric_range' % name)) del p # Multiple plots with same range. p = plot_parametric((sin(x), cos(x)), (x, sin(x))) p.save(tmp_file('%s_parametric_multiple' % name)) del p # Multiple plots with different ranges. p = plot_parametric((sin(x), cos(x), (x, -3, 3)), (x, sin(x), (x, -5, 5))) p.save(tmp_file('%s_parametric_multiple_ranges' % name)) del p # depth of recursion specified. p = plot_parametric(x, sin(x), depth=13) p.save(tmp_file('%s_recursion_depth' % name)) del p # No adaptive sampling. p = plot_parametric(cos(x), sin(x), adaptive=False, nb_of_points=500) p.save(tmp_file('%s_adaptive' % name)) del p # 3d parametric plots p = plot3d_parametric_line(sin(x), cos(x), x) assert str(p) == """Plot object containing: [0]: 3D parametric cartesian line: (sin(x), cos(x), x) for x over (-10.0, 10.0)""" p.save(tmp_file('%s_3d_line' % name)) del p p = plot3d_parametric_line((sin(x), cos(x), x, (x, -5, 5)), (cos(x), sin(x), x, (x, -3, 3))) p.save(tmp_file('%s_3d_line_multiple' % name)) del p p = plot3d_parametric_line(sin(x), cos(x), x, nb_of_points=30) p.save(tmp_file('%s_3d_line_points' % name)) del p # 3d surface single plot. p = plot3d(x * y) assert str(p) in ("""Plot object containing: [0]: cartesian surface: x*y for %s over (-10.0, 10.0) and %s over (-10.0, 10.0)""" % _ for _ in [(x, y), (y, x)]) p.save(tmp_file('%s_surface' % name)) del p # Multiple 3D plots with same range. p = plot3d(-x * y, x * y, (x, -5, 5)) p.save(tmp_file('%s_surface_multiple' % name)) del p # Multiple 3D plots with different ranges. p = plot3d((x * y, (x, -3, 3), (y, -3, 3)), (-x * y, (x, -3, 3), (y, -3, 3))) p.save(tmp_file('%s_surface_multiple_ranges' % name)) del p # Single Parametric 3D plot p = plot3d_parametric_surface(sin(x + y), cos(x - y), x - y) assert str(p) in ("""Plot object containing: [0]: parametric cartesian surface: (sin(x + y), cos(x - y), x - y) for %s over (-10.0, 10.0) and %s over (-10.0, 10.0)""" % _ for _ in [(x, y), (y, x)]) p.save(tmp_file('%s_parametric_surface' % name)) del p # Multiple Parametric 3D plots. p = plot3d_parametric_surface( (x * sin(z), x * cos(z), z, (x, -5, 5), (z, -5, 5)), (sin(x + y), cos(x - y), x - y, (x, -5, 5), (y, -5, 5))) p.save(tmp_file('%s_parametric_surface' % name)) del p finally: TmpFileManager.cleanup()
def test_matplotlib_intro(): """Examples from the 'introduction' notebook.""" try: name = 'test' tmp_file = TmpFileManager.tmp_file p = plot(x, adaptive=False) assert str(p) == """Plot object containing: [0]: cartesian line: x for x over (-10.0, 10.0)""" p = plot(x * sin(x), x * cos(x)) p.extend(p) p[0].line_color = lambda a: a p[1].line_color = 'b' p.title = 'Big title' p.xlabel = 'the x axis' p[1].label = 'straight line' p.aspect_ratio = (1, 1) p.xlim = (-15, 20) p.save(tmp_file(f'{name}_basic_options_and_colors')) p.extend(plot(x + 1)) p.append(plot(x + 3, x**2)[1]) p.save(tmp_file(f'{name}_plot_extend_append')) p[2] = plot(x**2, (x, -2, 3)) p.save(tmp_file(f'{name}_plot_setitem')) del p p = plot(sin(x), (x, -2 * pi, 4 * pi)) p.save(tmp_file(f'{name}_line_explicit')) del p p = plot(sin(x)) p.save(tmp_file(f'{name}_line_default_range')) del p p = plot((x**2, (x, -5, 5)), (x**3, (x, -3, 3))) p.save(tmp_file(f'{name}_line_multiple_range')) del p pytest.raises(ValueError, lambda: plot(x, y)) plot(sin(x), axis_center='center', show=False).save(tmp_file()) plot(sin(x), axis_center=(0.4, 0.5)).save(tmp_file()) plot(sin(x), axis_center=False).save(tmp_file()) plot(sin(x), axis=False).save(tmp_file()) plot(sin(x), margin=1).save(tmp_file()) # parametric 2d plots. # Single plot with default range. plot_parametric(sin(x), cos(x)).save(tmp_file()) plot_parametric(sin(x), cos(x), depth=7).save(tmp_file()) plot_parametric(sin(x), cos(x), show=False).save(tmp_file()) # Single plot with range. p = plot_parametric(sin(x), cos(x), (x, -5, 5)) assert str(p) == """Plot object containing: [0]: parametric cartesian line: (sin(x), cos(x)) for x over (-5.0, 5.0)""" p.save(tmp_file(f'{name}_parametric_range')) p[0].line_color = lambda a: a p.save(tmp_file(f'{name}_parametric_color')) del p # Multiple plots with same range. p = plot_parametric((sin(x), cos(x)), (x, sin(x))) p.save(tmp_file(f'{name}_parametric_multiple')) del p # Multiple plots with different ranges. p = plot_parametric((sin(x), cos(x), (x, -3, 3)), (x, sin(x), (x, -5, 5))) p.save(tmp_file(f'{name}_parametric_multiple_ranges')) del p # depth of recursion specified. p = plot_parametric(x, sin(x), depth=13) p.save(tmp_file(f'{name}_recursion_depth')) del p # No adaptive sampling. p = plot_parametric(cos(x), sin(x), adaptive=False, nb_of_points=500) p.save(tmp_file(f'{name}_adaptive')) del p # 3d parametric plots p = plot3d_parametric_line(sin(x), cos(x), x) assert str(p) == """Plot object containing: [0]: 3D parametric cartesian line: (sin(x), cos(x), x) for x over (-10.0, 10.0)""" p.save(tmp_file(f'{name}_3d_line')) del p plot3d_parametric_line(sin(x), cos(x), x, show=False).save(tmp_file()) p = plot3d_parametric_line((sin(x), cos(x), x, (x, -5, 5)), (cos(x), sin(x), x, (x, -3, 3))) p.save(tmp_file(f'{name}_3d_line_multiple')) del p p = plot3d_parametric_line(sin(x), cos(x), x, nb_of_points=30) p.save(tmp_file(f'{name}_3d_line_points')) p.extend(plot_parametric(cos(x), sin(x))) pytest.raises(ValueError, p.show) del p # 3d surface single plot. p = plot3d(x * y) assert str(p) in ("""Plot object containing: [0]: cartesian surface: x*y for %s over (-10.0, 10.0) and %s over (-10.0, 10.0)""" % _ for _ in [(x, y), (y, x)]) p.save(tmp_file(f'{name}_surface')) del p plot3d(x * y, show=False).save(tmp_file()) # Multiple 3D plots with same range. p = plot3d(-x * y, x * y, (x, -5, 5)) p.save(tmp_file(f'{name}_surface_multiple')) del p # Multiple 3D plots with different ranges. p = plot3d((x * y, (x, -3, 3), (y, -3, 3)), (-x * y, (x, -3, 3), (y, -3, 3))) p.save(tmp_file(f'{name}_surface_multiple_ranges')) del p # Single Parametric 3D plot p = plot3d_parametric_surface(sin(x + y), cos(x - y), x - y) assert str(p) in ("""Plot object containing: [0]: parametric cartesian surface: (sin(x + y), cos(x - y), x - y) for %s over (-10.0, 10.0) and %s over (-10.0, 10.0)""" % _ for _ in [(x, y), (y, x)]) p.save(tmp_file(f'{name}_parametric_surface')) del p plot3d_parametric_surface(sin(x + y), cos(x - y), x - y, show=False).save(tmp_file()) # Multiple Parametric 3D plots. p = plot3d_parametric_surface( (x * sin(z), x * cos(z), z, (x, -5, 5), (z, -5, 5)), (sin(x + y), cos(x - y), x - y, (x, -5, 5), (y, -5, 5))) p.save(tmp_file(f'{name}_parametric_surface')) del p # issue sympy/sympy#7140 p1 = plot(x) p2 = plot(x**2) # append a series p2.append(p1[0]) assert len(p2._series) == 2 with pytest.raises(TypeError): p1.append(p2) with pytest.raises(TypeError): p1.append(p2._series) # issue sympy/sympy#10925 f = Piecewise((-1, x < -1), (x, And(-1 <= x, x < 0)), (x**2, And(0 <= x, x < 1)), (x**3, True)) p = plot(f, (x, -3, 3)) p.save(tmp_file(f'{name}_10925')) del p finally: TmpFileManager.cleanup()