def testViter_complementV01(): import axis as ax import numpy as np a = ax.IrrepAxis("C2v", (4,0,1,2), np.ndarray) ma = MultiAxis((a, a)) assert( list(ma.iter_complement((0,))) == [(0, 0), (2, 2), (3, 3)] ) assert( list(ma.iter_complement((1,))) == [(2, 3), (3, 2)] ) assert( list(ma.iter_complement((2,))) == [(0, 2), (2, 0)] ) assert( list(ma.iter_complement((3,))) == [(0, 3), (3, 0)] )
def testViter_complementV02(): ma = MultiAxis(()) assert( list(ma.iter_complement('chicken')) == [()] )