Spaces:
Building
Building
File size: 1,194 Bytes
e3df007 |
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 |
# Copyright (c) Meta Platforms, Inc. and affiliates.
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
from animated_drawings.model.vectors import Vectors
import numpy as np
def test_initialize_with_tuple_or_list1():
v1 = Vectors((0, 1, 2))
assert np.array_equal(v1.vs, np.array([[0, 1, 2]]))
v2 = Vectors([3, 4, 5])
assert np.array_equal(v2.vs, np.array([[3, 4, 5]]))
def test_initialize2():
try:
Vectors('f') # type: ignore
except AssertionError:
return
assert False
def test_initialize_with_single_dimensional_array():
v1 = Vectors(np.array([0, 1, 2]))
assert np.array_equal(v1.vs, np.array([[0, 1, 2]]))
def test_div():
v1 = Vectors(np.array([0, 1, 2]))
assert np.array_equal((v1/2).vs, np.array([[0.0, 0.5, 1.0]]))
def test_norm():
v1 = Vectors(np.array([10, 10, 10]))
v1.norm()
v2 = Vectors(np.array([10, 10, 10]) / 300**0.5)
assert np.array_equal(v1.vs, v2.vs)
def test_norm_zero():
v1 = Vectors(np.array([0, 0, 0]))
v1.norm()
v2 = Vectors(np.array([0, 0, 0]) / 1**0.5)
assert np.array_equal(v1.vs, v2.vs)
|