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
|
# -*- 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 RegularPolygonVisual
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_regular_polygon_draw1():
"""Test drawing regular polygons without transforms using RegularPolygonVisual""" # noqa
with TestingCanvas() as c:
rpolygon = visuals.RegularPolygon(center=(0., 0.), radius=0.4, sides=8,
color=(1, 0, 0, 1),
parent=c.scene)
rpolygon.transform = transforms.STTransform(scale=(50, 50),
translate=(50, 50))
assert_image_approved(c.render(), 'visuals/regular_polygon1.png')
rpolygon.parent = None
rpolygon = visuals.RegularPolygon(center=(0., 0.), radius=0.4, sides=8,
color=(1, 0, 0, 1),
border_color=(0, 1, 1, 1),
parent=c.scene)
rpolygon.transform = transforms.STTransform(scale=(50, 50),
translate=(50, 50))
assert_image_approved(c.render(), 'visuals/regular_polygon2.png')
rpolygon.parent = None
rpolygon = visuals.RegularPolygon(center=(0., 0.), radius=0.4, sides=8,
border_color=(0, 1, 1, 1),
parent=c.scene)
rpolygon.transform = transforms.STTransform(scale=(50, 50),
translate=(50, 50))
assert_image_approved(c.render(), 'visuals/regular_polygon3.png',
min_corr=0.7)
@requires_application()
def test_regular_polygon_draw2():
"""Test drawing transformed regular polygons using RegularPolygonVisual""" # noqa
with TestingCanvas() as c:
rpolygon = visuals.RegularPolygon(center=(0., 0.), radius=0.4, sides=8,
color=(0, 0, 1, 1),
parent=c.scene)
rpolygon.transform = transforms.STTransform(scale=(75, 100),
translate=(50, 50))
assert_image_approved(c.render(), 'visuals/regular_polygon4.png')
rpolygon.parent = None
rpolygon = visuals.RegularPolygon(center=(0., 0.), radius=0.4, sides=8,
color=(0, 0, 1, 1),
border_color=(1, 0, 0, 1),
parent=c.scene)
rpolygon.transform = transforms.STTransform(scale=(75, 100),
translate=(50, 50))
assert_image_approved(c.render(), 'visuals/regular_polygon5.png')
rpolygon.parent = None
rpolygon = visuals.RegularPolygon(center=(0., 0.), radius=0.4, sides=8,
border_color=(1, 0, 0, 1),
parent=c.scene)
rpolygon.transform = transforms.STTransform(scale=(75, 100),
translate=(50, 50))
assert_image_approved(c.render(), 'visuals/regular_polygon6.png',
min_corr=0.6)
@requires_application()
def test_reactive_draw():
"""Test reactive regular polygon attributes"""
with TestingCanvas() as c:
rpolygon = visuals.RegularPolygon(center=[50, 50, 0.], radius=20,
sides=8,
color='yellow',
parent=c.scene)
rpolygon.center = [70, 40, 0.]
assert_image_approved(c.render(),
'visuals/reactive_regular_polygon1.png')
rpolygon.radius = 25
assert_image_approved(c.render(),
'visuals/reactive_regular_polygon2.png')
rpolygon.color = 'red'
assert_image_approved(c.render(),
'visuals/reactive_regular_polygon3.png')
rpolygon.border_color = 'yellow'
assert_image_approved(c.render(),
'visuals/reactive_regular_polygon4.png')
rpolygon.sides = 6
assert_image_approved(c.render(),
'visuals/reactive_regular_polygon5.png')
run_tests_if_main()
|
