122 lines
3.1 KiB
Python
122 lines
3.1 KiB
Python
from unittest import TestCase
|
|
|
|
from vector import Vector
|
|
|
|
|
|
class TestVector(TestCase):
|
|
def test_should_create_vector_dim_5(self):
|
|
vector = Vector(5)
|
|
|
|
actual = vector.get_dimension()
|
|
expected = 5
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_should_create_zero_vector(self):
|
|
vector = Vector(5)
|
|
|
|
actual = vector.get_data()
|
|
expected = [0, 0, 0, 0, 0]
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_should_create_vector(self):
|
|
data = list(range(5))
|
|
vector = Vector(data)
|
|
|
|
actual = vector.get_data()
|
|
expected = [0, 1, 2, 3, 4]
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_should_neg_vector(self):
|
|
v = Vector([1, 2])
|
|
|
|
expected = Vector([-1, -2])
|
|
actual = -v
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_should_add_vectors(self):
|
|
v1 = Vector([1, 2])
|
|
v2 = Vector([3, 4])
|
|
|
|
expected = Vector([4, 6])
|
|
actual = v1 + v2
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_should_add_scalar_to_vector(self):
|
|
v = Vector([1, 2])
|
|
s = 2
|
|
|
|
expected = Vector([3, 4])
|
|
actual = v + s
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_should_radd_scalar_to_vector(self):
|
|
v = Vector([1, 2])
|
|
s = 2
|
|
|
|
expected = Vector([3, 4])
|
|
actual = s + v
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_should_raise_value_missmatch_error_while_adding_vectors(self):
|
|
v1 = Vector(1)
|
|
v2 = Vector(2)
|
|
self.assertRaises(ValueError, lambda: v1 + v2)
|
|
|
|
def test_should_raise_dimension_error_while_adding_vectors(self):
|
|
v1 = Vector(1)
|
|
v2 = '0'
|
|
self.assertRaises(ValueError, lambda: v1 + v2)
|
|
|
|
def test_should_subtract_vectors(self):
|
|
v1 = Vector([1, 2])
|
|
v2 = Vector([3, 4])
|
|
|
|
expected = Vector([-2, -2])
|
|
actual = v1 - v2
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_should_subtract_scalar_of_vector(self):
|
|
v = Vector([1, 2])
|
|
s = 2
|
|
|
|
expected = Vector([-1, 0])
|
|
actual = v - s
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_should_rsubtract_scalar_of_vector(self):
|
|
v = Vector([1, 2])
|
|
s = 2
|
|
|
|
expected = Vector([1, 0])
|
|
actual = s - v
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_should_multiply_vectors(self):
|
|
v1 = Vector([1, 2])
|
|
v2 = Vector([3, 4])
|
|
|
|
expected = 11
|
|
actual = v1 * v2
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_should_multiply_scalar_with_vector(self):
|
|
v = Vector([1, 2])
|
|
s = 2
|
|
|
|
expected = Vector([2, 4])
|
|
actual = v * s
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_should_rmultiply_scalar_with_vector(self):
|
|
v = Vector([1, 2])
|
|
s = 2
|
|
|
|
expected = Vector([2, 4])
|
|
actual = s * v
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_should_raise_value_missmatch_error_while_multiplying_vectors(self):
|
|
v1 = Vector([1, 2])
|
|
v2 = '0'
|
|
self.assertRaises(ValueError, lambda: v1 * v2)
|