def testSetCoords(self):
v0 = Vector()
v0.setcoords([0, 0])
v1 = Vector()
v1.setcoords([2, 1])
self.assertEqual(v0.vekt, [0, 0])
assert v1.vekt == [2, 1]
def mbm_derivatives(self, atom, odl):
lista_sil = [0 for _ in xrange(atom.wymiar)]
for i in range(len(odl)):
lista_sil[i] += -self.a * 2 * self.b * (odl[i] - 1) * (e ** (-self.b * ((odl[i] - 1) ** 2))) \
- self.c * (odl[i] + 1) * e ** (-(odl[i] + 1) ** 2) \
- self.d * 4 * atom.pol.vekt[i] ** 3
sily = Vector()
sily.setcoords(lista_sil)
atom.przysp += sily
def testRownosciWektorow(self):
v0 = Vector()
v1 = Vector()
v1.setcoords([1, 2])
v2 = Vector()
v2.setcoords([1, 2])
self.assertNotEquals(v0, v1)
self.assertEqual(v1, v2)
self.assertEqual(v0, v0)
def miekkie_derivatives(self, atom):
lista_sil = []
for p in atom.pol:
if p > self.l:
lista_sil.append(self.f * p * (self.l - p) / p)
elif p < -self.l:
lista_sil.append(self.f * p * (self.l + p) / abs(p))
else:
lista_sil.append(0.0)
sily = Vector()
sily.setcoords(lista_sil)
atom.przysp += sily
def testOdejmowanieWektorow(self):
v0 = Vector()
v0.setcoords([3, 3])
v1 = Vector()
v1.setcoords([1, 2])
v2 = Vector()
v2.setcoords([2, 1])
self.assertEqual(v2, v0 - v1)
self.assertEqual(v0.vekt, [3, 3])
def testDodawanieWektorow(self):
v0 = Vector()
v0.setcoords([2, 1])
v1 = Vector()
v1.setcoords([1, 2])
v3 = Vector()
v3.setcoords([3, 3])
self.assertEqual(v3, v1 + v0)
self.assertEqual(v1.vekt, [1, 2])
def testIsInstance(self):
v1 = Vector()
v2 = Vector()
v1.setcoords([0, 0, 0])
v2.setcoords([0.1, 0.1, 0.1])
wynik = Vector()
wynik.setcoords([0.1, 0.1, 0.1])
assert isinstance(v1, Vector)
assert isinstance(v2, Vector)
assert v1 + v2 == wynik
def testMozenieWektorow(self):
skalar = 5
v0 = Vector()
v0.setcoords([1, 1])
v1 = Vector()
v1.setcoords([2, 2])
v2 = Vector()
v2.setcoords([5, 5])
v3 = 4
self.assertEqual(v3, v1 * v0)
self.assertEqual(v2, skalar * v0)
def testLenghtWektorow(self):
v0 = Vector()
v0.setcoords([3, 2])
self.assertAlmostEqual(v0.lenght(), (13) ** 0.5, delta=0.0000001)
def testDlugosciWektorow(self):
v1 = Vector()
v1.setcoords([1, 1])
self.assertEqual(v1.lenght(), 2.0 ** (1.0 / 2))
