def test_distance(self): v1 = Vec(1, 0, 0) v2 = Vec(0, 1, 0) dist = v1.distance(v2) correct = sqrt(2) self.assertEqual(dist, correct)