PythonPerformance_RayTracing/tests/test_vector.py at parallel · PierreRust/PythonPerformance_RayTracing · GitHub

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
from .context import raytracer

from raytracer import vector
from raytracer.vector import Vector3


def test_norm():
    v = Vector3(1, 0, 0)
    assert v.norm() == 1

    v = Vector3(0, 5, 0)
    assert v.norm() == 5


def test_normalize():
    v = Vector3(2, 0, 0)
    assert v.normalize() == Vector3(1, 0, 0)


def test_normalize_already_norm():
    v = Vector3(1, 0, 0)
    assert v.normalize() == v


def test_cross_product():
    v1 = Vector3(1, 0, 0)
    v2 = Vector3(0, 1, 0)

    assert v1.cross(v2) == Vector3(0, 0, 1)
    assert v2.cross(v1) == Vector3(0, 0, -1)


def test_dot_product():
    v1 = Vector3(1, 0, 2)
    v2 = Vector3(2, 3, 4)

    assert v1.dot(v2) == 10
    assert v2.dot(v1) == 10


def test_scalar_multiplication():
    v1 = Vector3(1, 0, 2)

    assert v1 * 3 == Vector3(3, 0, 6)
    assert 3 * v1 == Vector3(3, 0, 6)


def test_element_wise_multiplication():
    v1 = Vector3(1, 0, 2)
    v2 = Vector3(2, 3, 4)

    assert v1 * v2 == Vector3(2, 0, 8)
    assert v2 * v1 == Vector3(2, 0, 8)


def test_scalar_addition():
    v1 = Vector3(1, 6, -1)

    obtained = v1 + 2
    assert obtained == Vector3(3, 8, 1)

    obtained = 2 + v1
    assert obtained == Vector3(3, 8, 1)


def test_element_wise_addition():
    v1 = Vector3(1, 6, -1)
    v2 = Vector3(0, 1, 2)

    obtained = v1 + v2
    assert obtained == Vector3(1, 7, 1)

    obtained = v2 + v1
    assert obtained == Vector3(1, 7, 1)


def test_scalar_subtraction():
    v1 = Vector3(1, 6, -1)
    obtained = v1 - 1
    assert obtained == Vector3(0, 5, -2)

    obtained = 1 - v1
    assert obtained == Vector3(0, -5, 2)


def test_element_wise_subtraction():
    v1 = Vector3(1, 6, -1)
    v2 = Vector3(0, 1, 2)
    obtained = v2 - v1
    assert obtained == Vector3(-1, -5, 3)


def test_scalar_division():
    v1 = Vector3(1, 6, -1)
    obtained = v1 / 2
    assert obtained == Vector3(0.5, 3, -0.5)


def test_element_wise_division():
    v1 = Vector3(1, 6, -1)
    v2 = Vector3(1, 2, -0.5)

    obtained = v1 / v2
    assert obtained == Vector3(1, 3, 2)


def test_copy_create():
    v1 = Vector3(1, 6, -1)
    v2 = Vector3(*v1)
    assert id(v1) != id(v2)
    assert v1 == v2