def test_item_errors():
vec = Vector.from_size(3)
with pytest.raises(TypeError):
vec["x"]
with pytest.raises(TypeError):
vec["x"] = 5
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 test_attrs():
vec = Vector.from_size(4)
assert vec.x == 0
vec.x = 1
assert vec.data == [1, 0, 0, 0]
assert vec.yx.data == [0, 1]
vec.yz = 42
assert vec.data == [1, 42, 42, 0]
assert vec.i == vec.x
assert vec.zwxyyy.data == [42, 0, 1, 42, 42, 42]
def test_items():
vec = Vector.from_size(5)
assert vec[0] == 0
vec[0] = 1
assert vec.data == [1, 0, 0, 0, 0]
assert vec[1, 0].data == [0, 1]
vec[1, 2] = 42
assert vec.data == [1, 42, 42, 0, 0]
assert vec[1:-1:2].data == [42, 0]
vec[1:-1] = 2 # range(*item.indices(len(self)))
assert vec.data == [1, 2, 2, 2, 0]
