def test_init():
# note that we try to ONLY test init here when possible, so we do things like comparing data attributes to a list
# constructing using both args and kwargs here should give us the same result
assert ([1, 2, 3] ==
Vector(1, 2, 3).data ==
Vector(x=1, y=2, z=3).data ==
Vector(1, z=3, y=2).data)
# same as kwargs, but tests that skipped-over places are filled with 0s
assert Vector(z=4, w=2).data == [0, 0, 4, 2]
# now we try a few combinations of creating a vector from some iterable
assert ([1, 2, 3] ==
Vector([1, 2, 3]).data ==
Vector(iter([1, 2, 3])).data ==
Vector.from_iter([1, 2, 3]).data ==
Vector.from_iter(iter([1, 2, 3])).data)
# now check from_size construction
from_size = Vector.from_size(2)
assert from_size.data == [0, 0]
assert Vector.from_size(5, fill=42).data == [42, 42, 42, 42, 42]
# try connecting a bunch of things together
from_connect = Vector(from_size, 1, Vector(1, 2), [3, 5])
assert from_connect.data == [0, 0, 1, 1, 2, 3, 5]
# verify that from_size wasn't somehow mutated in the process
assert from_size.data == [0, 0]
# now lets create some vectors using from_polar
from_polar = Vector.from_polar(42, math.pi, 0, 0)
from_polar_3d_units = [
Vector.from_polar(1, 0, 0), # Vector(1, 0, 0)
Vector.from_polar(1, math.pi/2, 0), # Vector(0, 1, 0)
Vector.from_polar(1, math.pi/2, math.pi/2) # Vector(0, 0, 1)
]
# we won't get very accurate results, so we need to check aproximate equality
assert all(map(aprox_equal, from_polar.data, [-42, 0, 0, 0]))
for dim, unit in enumerate(from_polar_3d_units):
correct_unit = Vector.from_size(3).data
correct_unit[dim] = 1
assert all(map(aprox_equal, unit.data, correct_unit))
def init_vs_from_iter():
init_list = [1, 2, 3]
print("__init__:", timeit(lambda: Vector(init_list)))
print("from_iter:", timeit(lambda: Vector.from_iter(init_list)))
