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))