def test_components(self):
self.assertEqual(Vector(1, 2).components, [1, 2])
def test_mul_with_vector(self):
self.assertEqual(Vector(1, 2) * Vector(2, 3), Vector(2, 6))
def test_mul_incompatible_type_error(self):
with self.assertRaises(TypeError):
Vector(1, 2) * 'incompatible'
def test_sub_with_vector(self):
self.assertEqual(Vector(3, 5) - Vector(2, 3), Vector(1, 2))
def test_mul_with_int(self):
self.assertEqual(Vector(1, 2) * 2, Vector(2, 4))
def test_add_incompatible_vector_error(self):
with self.assertRaises(TypeError):
Vector(1, 2) + Vector(1, 2, 3)
def test_sub_with_int(self):
self.assertEqual(Vector(1, 2) - 1, Vector(0, 1))
def test_contract(self):
self.assertEqual(Vector(3, 2).contract(Vector(1, 3)), Vector(1, 2))
def test_contract_incompatible_type_error(self):
with self.assertRaises(TypeError):
Vector(3, 2).contract('incompatible')
def test_expand_incompatible_type_error(self):
with self.assertRaises(TypeError):
Vector(1, 2).expand('incompatible')
def test_expand_incompatible_vector_error(self):
with self.assertRaises(TypeError):
Vector(1, 2).expand(Vector(3, 1, 0))
def test_expand(self):
self.assertEqual(Vector(1, 2).expand(Vector(3, 1)), Vector(3, 2))
def test_magnitude(self):
self.assertEqual(Vector(2, 3, 0.5).magnitude, 3)
def test_to_int_list(self):
self.assertEqual(Vector(1, 2.5).to_int_list(), [1, 2])
def test_add_with_float(self):
self.assertEqual(Vector(1, 2) + 0.5, Vector(1.5, 2.5))
def test_contract_incompatible_vector_error(self):
with self.assertRaises(TypeError):
Vector(3, 2).contract(Vector(1, 2, 0))
def test_add_with_vector(self):
self.assertEqual(Vector(1, 2) + Vector(2, 3), Vector(3, 5))
def test_to_float_list(self):
self.assertEqual(Vector(1, 2.5).to_float_list(), [1.0, 2.5])
def test_repr(self):
self.assertEqual(repr(Vector(1, 2)), 'Vector(1.0, 2.0)')
def test_eq(self):
self.assertEqual(Vector(1, 2), Vector(1.0, 2.0))
def test_sub_with_float(self):
self.assertEqual(Vector(1, 2) - 0.5, Vector(0.5, 1.5))
def test_really_not_eq(self):
self.assertNotEqual(Vector(1, 2), Vector(2.0, 1.0))
def test_sub_incompatible_vector_error(self):
with self.assertRaises(TypeError):
Vector(3, 5) - Vector(2, 3, 4)
def test_definitely_not_eq(self):
self.assertNotEqual(Vector(1, 2), Vector(1, 2, 3))
def test_mul_with_float(self):
self.assertEqual(Vector(1, 2) * 0.5, Vector(0.5, 1))
def test_really_definitely_not_eq(self):
self.assertNotEqual(Vector(1, 2), 'incompatible')
def test_str(self):
self.assertEqual(str(Vector(1, 2)), '1 2')
def test_add_with_int(self):
self.assertEqual(Vector(1, 2) + 1, Vector(2, 3))
def test_mul_incompatible_vector_error(self):
with self.assertRaises(TypeError):
Vector(1, 2) * Vector(2, 3, 4)
def test_iter(self):
self.assertEqual([c for c in Vector(1, 2, 3)], [1, 2, 3])