python vec_class Vector 的创建

创建一个向量的class,可以让vector相加,相减, 乘以一个系数, 或者清空成为(0,0,0)

from math import sqrt

class Vector:

"""

A Vector is a 3-tuple of (x,y,z) coordinates.

"""

def __init__(self,x,y,z):

self._x = x

self._y = y

self._z = z

def __repr__(self):

return '{%.3g} ,{%.3g} ,{%.3g}'%(self._x, self._y, self._z)

def __str__(self):

return '({},{},{})'.format(self._x, self._y, self._z)

def __add__(self,other):

return Vector(self._x + other._x, self._y + other._y, self._z + other._z)

def __sub__(self,other):

return Vector(self._x - other._x, self._y - other._y, self._z - other._z)

def norm(self):

result = sqrt(self._x**2 + self._y**2 + self._z**2)

return result

def __mul__(self,other):

return Vector(self._x * other, self * other, self._z * other)

def x(self):

return self._x

def y(self):

return self._y

def z(self):

return self._z

def clear(self):

return Vector(0 ,0 ,0)

以下是我test的结果

v1 = Vector(1, 1, 1)

v2 = Vector(2, 2, 2)

v3 = Vector(3, 3, 3)

assert str(v1) == '(1,1,1)'

assert str(v2) == '(2,2,2)'

assert Vector(1, 2, 3) == Vector(1, 2, 3)

assert Vector(1, 2, 3) != Vector(1.001, 2.001, 3.001)

---------------------------------------------------------------------------

AssertionError Traceback (most recent call last)

in ()

----> 1 assert Vector(1, 2, 3) == Vector(1, 2, 3)

2 assert Vector(1, 2, 3) != Vector(1.001, 2.001, 3.001)

AssertionError:

assert v1 + v2 == v3

---------------------------------------------------------------------------

AssertionError Traceback (most recent call last)

in ()

----> 1 assert v1 + v2 == v3

AssertionError:

assert v3 - v2 == v1

---------------------------------------------------------------------------

AssertionError Traceback (most recent call last)

in ()

----> 1 assert v3 - v2 == v1

AssertionError:

assert v3 == v1 * 3

#Error 信息太多 不复制出来了

assert round(v2.norm(), 10) == round(sqrt(12), 10)

v1.clear()

assert v1 == Vector(0,0,0)

---------------------------------------------------------------------------

AssertionError Traceback (most recent call last)

in ()

1 v1.clear()

----> 2 assert v1 == Vector(0,0,0)

AssertionError:

问题到底出在哪里?

比如向量相加,我print v1+v2 是(3,3,3)v3 也是(3,3,3),为什么还是assertion error?

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值