def test_init(self):
m = balancer.mass(5, 30)
self.assertEqual(m.weight, 5)
self.assertEqual(m.angle, 30)
m = balancer.mass()
self.assertEqual(m.weight, 0)
self.assertEqual(m.angle, 0)
def test_calc(self):
coeff = balancer.coeff()
coeff.v0 = balancer.vib(2, 2)
coeff.mt = balancer.mass(5, 1)
k = coeff.calc()
def test_basic(self):
unbalance = balancer.mass(4, 90)
rotor = balancer.rotor(unbalance)
rotor.calc()