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
|
from unittest.mock import MagicMock, call
import pytest
from PyQt6.QtOpenGL import QOpenGLShader
from UM.View.GL.ShaderProgram import ShaderProgram, InvalidShaderProgramError
import os
def test_ShaderProgramInit():
# Creating it just shouldn't fail.
shader = ShaderProgram()
def test_loadEmpty():
shader = ShaderProgram()
with pytest.raises(InvalidShaderProgramError):
shader.load(os.path.join(os.path.dirname(os.path.abspath(__file__)), "Shaders", "empty.shader"))
def test_loadInvalid():
shader = ShaderProgram()
with pytest.raises(InvalidShaderProgramError):
shader.load(os.path.join(os.path.dirname(os.path.abspath(__file__)), "Shaders", "invalid.shader"))
with pytest.raises(InvalidShaderProgramError):
shader.load(os.path.join(os.path.dirname(os.path.abspath(__file__)), "Shaders", "invalid2.shader"))
def test_load():
shader = ShaderProgram()
mocked_shader_program = MagicMock()
shader._shader_program = mocked_shader_program
shader.load(os.path.join(os.path.dirname(os.path.abspath(__file__)), "Shaders", "test.shader"))
# It should be called 3 times, once for vertex, once for fragment and once for geometry
call_arg_list = mocked_shader_program.addShaderFromSourceCode.call_args_list
assert call(QOpenGLShader.Vertex, "vertex_code") in call_arg_list
assert call(QOpenGLShader.Fragment, "fragment_code") in call_arg_list
assert call(QOpenGLShader.Geometry, "geometry_code") in call_arg_list
def test_bindAndRelease():
shader = ShaderProgram()
mocked_shader_program = MagicMock()
shader._shader_program = mocked_shader_program
shader.bind()
assert mocked_shader_program.bind.call_count == 1
# Doing it without releasing in between shouldn't cause another bind.
shader.bind()
assert mocked_shader_program.bind.call_count == 1
shader.release()
assert mocked_shader_program.release.call_count == 1
shader.release()
assert mocked_shader_program.release.call_count == 1
# We left it unbound, so now binding should work.
shader.bind()
assert mocked_shader_program.bind.call_count == 2
enable_attribute_data = [("int", 2, 3, 0x1404, 1),
("float", 5, 9, 0x1406, 1),
("vector2f", 2, 32, 0x1406, 2),
("vector3f", 2000, 90, 0x1406, 3),
("vector4f", 12, 1, 0x1406, 4)]
@pytest.mark.parametrize("type, offset, stride, type_result, tuple_size", enable_attribute_data)
def test_enableDisableAttribute(type, offset, stride, type_result, tuple_size):
shader = ShaderProgram()
mocked_shader_program = MagicMock()
shader._shader_program = mocked_shader_program
attribute_index = MagicMock()
mocked_shader_program.attributeLocation = MagicMock(return_value = attribute_index)
shader.enableAttribute("blorp", type, offset, stride)
# Validate
mocked_shader_program.setAttributeBuffer.assert_called_once_with(attribute_index, type_result, offset, tuple_size, stride)
mocked_shader_program.enableAttributeArray.assert_called_once_with(attribute_index)
# Disable it again
shader.disableAttribute("blorp")
mocked_shader_program.disableAttributeArray.assert_called_once_with(attribute_index)
# Disable unknown attribute
shader.disableAttribute("BEEP")
mocked_shader_program.disableAttributeArray.assert_called_once_with(attribute_index)
|
