1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
# -*- coding: utf-8 -*-
# Copyright (c) Vispy Development Team. All Rights Reserved.
# Distributed under the (new) BSD License. See LICENSE.txt for more info.
"""
Tests for EllipseVisual
All images are of size (100,100) to keep a small file size
"""
from vispy.scene import visuals, transforms
from vispy.testing import (requires_application, TestingCanvas,
run_tests_if_main)
from vispy.testing.image_tester import assert_image_approved
@requires_application()
def test_circle_draw():
"""Test drawing circles without transform using EllipseVisual"""
with TestingCanvas() as c:
ellipse = visuals.Ellipse(center=(75, 35, 0), radius=20,
color=(1, 0, 0, 1),
parent=c.scene)
assert_image_approved(c.render(), 'visuals/circle1.png')
ellipse.parent = None
ellipse = visuals.Ellipse(center=(75, 35, 0), radius=20,
color=(1, 0, 0, 1),
border_color=(0, 1, 1, 1),
parent=c.scene)
assert_image_approved(c.render(), 'visuals/circle2.png')
ellipse.parent = None
ellipse = visuals.Ellipse(center=(75, 35, 0), radius=20,
border_color=(0, 1, 1, 1),
parent=c.scene)
# low corr here because borders have some variability
# esp. w/HiDPI
assert_image_approved(c.render(), 'visuals/circle3.png',
min_corr=0.7)
@requires_application()
def test_ellipse_draw():
"""Test drawing transformed ellipses using EllipseVisual"""
with TestingCanvas() as c:
ellipse = visuals.Ellipse(center=(0., 0.), radius=(20, 15),
color=(0, 0, 1, 1),
parent=c.scene)
ellipse.transform = transforms.STTransform(scale=(2.0, 3.0),
translate=(50, 50))
assert_image_approved(c.render(), 'visuals/ellipse1.png')
ellipse.parent = None
ellipse = visuals.Ellipse(center=(0., 0.), radius=(20, 15),
color=(0, 0, 1, 1),
border_color=(1, 0, 0, 1),
parent=c.scene)
ellipse.transform = transforms.STTransform(scale=(2.0, 3.0),
translate=(50, 50))
assert_image_approved(c.render(), 'visuals/ellipse2.png')
ellipse.parent = None
ellipse = visuals.Ellipse(center=(0., 0.), radius=(20, 15),
border_color=(1, 0, 0, 1),
parent=c.scene)
ellipse.transform = transforms.STTransform(scale=(2.0, 3.0),
translate=(50, 50))
assert_image_approved(c.render(), 'visuals/ellipse3.png',
min_corr=0.7)
@requires_application()
def test_arc_draw1():
"""Test drawing arcs using EllipseVisual"""
with TestingCanvas() as c:
ellipse = visuals.Ellipse(center=(50., 50.), radius=(20, 15),
start_angle=150., span_angle=120.,
color=(0, 0, 1, 1),
parent=c.scene)
assert_image_approved(c.render(), 'visuals/arc1.png')
ellipse.parent = None
ellipse = visuals.Ellipse(center=(50., 50.), radius=(20, 15),
start_angle=150., span_angle=120.,
border_color=(1, 0, 0, 1),
parent=c.scene)
assert_image_approved(c.render(), 'visuals/arc2.png',
min_corr=0.6)
@requires_application()
def test_reactive_draw():
"""Test reactive ellipse attributes"""
with TestingCanvas() as c:
ellipse = visuals.Ellipse(center=[75, 35, 0.], radius=[20, 15],
color='yellow',
parent=c.scene)
ellipse.center = [70, 40, 0.]
assert_image_approved(c.render(), 'visuals/reactive_ellipse1.png')
ellipse.radius = 25
assert_image_approved(c.render(), 'visuals/reactive_ellipse2.png')
ellipse.color = 'red'
assert_image_approved(c.render(), 'visuals/reactive_ellipse3.png')
ellipse.border_color = 'yellow'
assert_image_approved(c.render(), 'visuals/reactive_ellipse4.png')
ellipse.start_angle = 140.
assert_image_approved(c.render(), 'visuals/reactive_ellipse5.png')
ellipse.span_angle = 100.
assert_image_approved(c.render(), 'visuals/reactive_ellipse6.png')
ellipse.num_segments = 10.
assert_image_approved(c.render(), 'visuals/reactive_ellipse7.png')
run_tests_if_main()
|
