diff --git a/libs/pyglet_rs/src/test/vector.py b/libs/pyglet_rs/src/test/vector.py index 7ede6ce..3168d07 100644 --- a/libs/pyglet_rs/src/test/vector.py +++ b/libs/pyglet_rs/src/test/vector.py @@ -7,6 +7,8 @@ import random import unittest +from typing import Tuple + from libs.pyglet_rs import Vector2_rs, Vector3_rs, Vector4_rs @@ -20,12 +22,18 @@ def print_test_info(func): return wrapper +def gen_random_vector() -> Tuple[Vector2_rs, Vector3_rs, Vector4_rs]: + vec2 = Vector2_rs(random.randint(1, 100), random.randint(1, 100)) + vec3 = Vector3_rs(random.randint(1, 100), random.randint(1, 100), random.randint(1, 100)) + vec4 = Vector4_rs(random.randint(1, 100), random.randint(1, 100), random.randint(1, 100), random.randint(1, 100)) + return vec2, vec3, vec4 + + class TestVector(unittest.TestCase): + @print_test_info def test1_create_print_vector(self): - vec2 = Vector2_rs(1, 2) - vec3 = Vector3_rs(1, 2, 3) - vec4 = Vector4_rs(1, 2, 3, 4) + vec2, vec3, vec4 = gen_random_vector() print(f"{vec2=}") print(f"{vec3=}") @@ -33,13 +41,8 @@ class TestVector(unittest.TestCase): @print_test_info def test2_calculate_vector(self): - vec2 = Vector2_rs(random.randint(1, 100), random.randint(1, 100)) - vec3 = Vector3_rs(random.randint(1, 100), random.randint(1, 100), random.randint(1, 100)) - vec4 = Vector4_rs(random.randint(1, 100), random.randint(1, 100), random.randint(1, 100), random.randint(1, 100)) - - vec2_1 = Vector2_rs(random.randint(1, 100), random.randint(1, 100)) - vec3_1 = Vector3_rs(random.randint(1, 100), random.randint(1, 100), random.randint(1, 100)) - vec4_1 = Vector4_rs(random.randint(1, 100), random.randint(1, 100), random.randint(1, 100), random.randint(1, 100)) + vec2, vec3, vec4 = gen_random_vector() + vec2_1, vec3_1, vec4_1 = gen_random_vector() print(f"{vec2=} {vec2_1=}") print(f"{vec3=} {vec3_1=}")