Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-11-10 13:40:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

update 80_linalg.py

Browse Source
This commit is contained in:
zhs628 2023-08-03 14:52:16 +00:00
parent 9f61f676a1
commit f00420aeda
1 changed files with 158 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

168
tests/80_linalg.py
View File

@ -1,7 +1,91 @@
from linalg import mat3x3, vec2, vec3
from linalg import mat3x3, vec2, vec3, vec4
import random
import sys
import math
# test vec2--------------------------------------------------------------------
def rotated_vec2(vec_2, radians: float):
cos_theta = math.cos(radians)
sin_theta = math.sin(radians)
new_x = vec_2.x * cos_theta - vec_2.y * sin_theta
new_y = vec_2.x * sin_theta + vec_2.y * cos_theta
return vec2(new_x, new_y)
# 生成随机测试目标
min_num = -10.0
max_num = 10.0
test_vec2 = vec2(*tuple([random.uniform(min_num, max_num) for _ in range(2)]))
static_test_vec2_float = vec2(3.1886954323, -1098399.59932453432)
static_test_vec2_int = vec2(278, -13919730938747)
# test __repr__
assert str(static_test_vec2_float) == 'vec2(3.1887, -1.0984e+06)'
assert str(static_test_vec2_int) == 'vec2(278, -1.39197e+13)'
# test copy
element_name_list = [e for e in dir(test_vec2) if e in 'x,y,z,w']
element_value_list = [getattr(test_vec2, attr) for attr in element_name_list]
copy_element_value_list = [getattr(test_vec2.copy(), attr) for attr in element_name_list]
assert element_value_list == copy_element_value_list
# test rotate
test_vec2_copy = test_vec2.copy()
radians = random.uniform(-10*math.pi, 10*math.pi)
test_vec2_copy = rotated_vec2(test_vec2_copy, radians)
assert test_vec2.rotate(radians).__dict__ == test_vec2_copy.__dict__
# test vec3--------------------------------------------------------------------
# 生成随机测试目标
min_num = -10.0
max_num = 10.0
test_vec3 = vec3(*tuple([random.uniform(min_num, max_num) for _ in range(3)]))
static_test_vec3_float = vec3(3.1886954323, -1098399.59932453432, 9.00000000000002765)
static_test_vec3_int = vec3(278, -13919730938747, 1364223456756456)
# test __repr__
assert str(static_test_vec3_float) == 'vec3(3.1887, -1.0984e+06, 9)'
assert str(static_test_vec3_int) == 'vec3(278, -1.39197e+13, 1.36422e+15)'
# test __getnewargs__
element_name_list = [e for e in dir(test_vec3) if e in 'x,y,z,w']
element_value_list = [getattr(test_vec3, attr) for attr in element_name_list]
assert tuple(element_value_list) == test_vec3.__getnewargs__()
# test copy
element_name_list = [e for e in dir(test_vec3) if e in 'x,y,z,w']
element_value_list = [getattr(test_vec3, attr) for attr in element_name_list]
copy_element_value_list = [getattr(test_vec3.copy(), attr) for attr in element_name_list]
assert element_value_list == copy_element_value_list
# test vec4--------------------------------------------------------------------
# 生成随机测试目标
min_num = -10.0
max_num = 10.0
test_vec4 = vec4(*tuple([random.uniform(min_num, max_num) for _ in range(4)]))
static_test_vec4_float = vec4(3.1886954323, -1098399.59932453432, 9.00000000000002765, 4565400000000.0000000045)
static_test_vec4_int = vec4(278, -13919730938747, 1364223456756456, -37)
# test __repr__
assert str(static_test_vec4_float) == 'vec4(3.1887, -1.0984e+06, 9, 4.5654e+12)'
assert str(static_test_vec4_int) == 'vec4(278, -1.39197e+13, 1.36422e+15, -37)'
# test __getnewargs__
element_name_list = [e for e in dir(test_vec4) if e in 'x,y,z,w']
element_value_list = [getattr(test_vec4, attr) for attr in element_name_list]
assert tuple(element_value_list) == test_vec4.__getnewargs__()
# test copy
element_name_list = [e for e in dir(test_vec4) if e in 'x,y,z,w']
element_value_list = [getattr(test_vec4, attr) for attr in element_name_list]
copy_element_value_list = [getattr(test_vec4.copy(), attr) for attr in element_name_list]
assert element_value_list == copy_element_value_list
# test mat3x3--------------------------------------------------------------------
def mat_round(mat, pos):
'''
对mat的副本的每一个元素执行round(element, pos)，返回副本
@ -39,11 +123,17 @@ def get_col(mat, col_index):
min_num = -10.0
max_num = 10.0
test_mat = mat3x3([[random.uniform(min_num, max_num) for _ in range(3)] for _ in range(3)])
# test_mat = mat3x3([
# [1, 2, 3],
# [4, 5, 6],
# [7, 8, 9]]
# )
static_test_mat_float= mat3x3([
[7.264189733952545, -5.432187523625671, 1.8765304152872613],
[-2.4910524352374734, 8.989660807513068, -0.7168824333280513],
[9.558042327611506, -3.336280256662496, 4.951381528057387]]
)
static_test_mat_int = mat3x3([
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
)
# test incorrect number of parameters is passed
for i in range(20):
@ -95,6 +185,18 @@ element_name_list = [e for e in dir(test_mat) if e[:2] != '__' and e[0] == '_']
for i, element in enumerate([getattr(test_mat, e) for e in element_name_list]):
assert test_mat.__getitem__((int(i/3), i%3)) == element
try:
test_mat[1,2,3]
raise Exception('未能触发错误拦截, 此处应当报错 IndexError("index out of range")')
except:
pass
try:
test_mat[-1][4]
raise Exception('未能触发错误拦截, 此处应当报错 IndexError("index out of range")')
except:
pass
# test __setitem__
test_mat_copy = test_mat.copy()
element_name_list = [e for e in dir(test_mat_copy) if e[:2] != '__' and e[0] == '_']
@ -102,6 +204,18 @@ for i, element in enumerate([getattr(test_mat_copy, e) for e in element_name_lis
test_mat_copy.__setitem__((int(i/3), i%3), list(range(9))[i])
assert test_mat_copy == mat3x3([[0,1,2], [3,4,5], [6,7,8]])
try:
test_mat[1,2,3] = 1
raise Exception('未能触发错误拦截, 此处应当报错 TypeError("Mat3x3.__setitem__ takes a tuple of 2 integers")')
except:
pass
try:
test_mat[-1][4] = 1
raise Exception('未能触发错误拦截, 此处应当报错 IndexError("index out of range")')
except:
pass
# test __add__
test_mat_copy = test_mat.copy()
ones = mat3x3()
@ -152,9 +266,43 @@ for i in range(3):
for j in range(3):
test_mat_copy[i, j] = test_mat[j, i]
# # test __repr__
# test_mat_copy = test_mat.copy()
# print(test_mat_copy[0,0])
# assert test_mat_copy.__repr__() == f'mat3x3([[{test_mat_copy[0,0].round(4)}, {test_mat_copy[1,0].round(4)}, {test_mat_copy[2,0].round(4)}],\n [{test_mat_copy[0,1].round(4)}, {test_mat_copy[1,1].round(4)}, {test_mat_copy[2,1].round(4)}],\n [{test_mat_copy[0,2].round(4)}, {test_mat_copy[1,2].round(4)}, {test_mat_copy[2,2].round(4)}]])'
# test __repr__
assert str(static_test_mat_float) == 'mat3x3([[7.2642, -5.4322, 1.8765],\n [-2.4911, 8.9897, -0.7169],\n [9.5580, -3.3363, 4.9514]])'
assert str(static_test_mat_int) == 'mat3x3([[1.0000, 2.0000, 3.0000],\n [4.0000, 5.0000, 6.0000],\n [7.0000, 8.0000, 9.0000]])'
# test __getnewargs__
test_mat_copy = test_mat.copy()
element_name_list = [e for e in dir(test_mat_copy) if e[:2] != '__' and e[0] == '_']
element_value_list = [getattr(test_mat, attr) for attr in element_name_list]
assert tuple(element_value_list) == test_mat.__getnewargs__()
# test __truediv__
test_mat_copy = test_mat.copy()
result_mat = test_mat_copy.__truediv__(12.345)
correct_result_mat = test_mat_copy.copy()
for i in range(3):
for j in range(3):
correct_result_mat[i, j] /= 12.345
assert result_mat == correct_result_mat
# test __rmul__
test_mat_copy = test_mat.copy()
result_mat = 12.345 * test_mat_copy
correct_result_mat = test_mat_copy.copy()
for i in range(3):
for j in range(3):
correct_result_mat[i, j] *= 12.345
assert result_mat == correct_result_mat
# 此处测试不完全, 未验证正确性
# test interface of "@" "matmul" "__matmul__" with vec3 and error handling
test_mat_copy = test_mat.copy()
test_mat_copy @ vec3(83,-9.12, 0.2983)
try:
test_mat_copy @ 12345
raise Exception('未能拦截错误 BinaryOptError("@") 在处理表达式 test_mat_copy @ 12345')