def test_setModVec_accepts_v3d(self): unit = UnitCell(3, 3, 3) vectors = (V3D(1, 2, 3), V3D(2, 1, 3), V3D(3, 2, 1)) unit.setModVec1(vectors[0]) unit.setModVec2(vectors[1]) unit.setModVec3(vectors[2]) for index, expected in enumerate(vectors): self.assertEquals(vectors[index], unit.getModVec(index))