Exemplos de MultiAxis.iter_complement em Python

Exemplo n.º 1

Arquivo: test_multiaxis.py Projeto: avcopan/spinblock

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

Exemplo n.º 2

Arquivo: test_multiaxis.py Projeto: avcopan/spinblock

def testViter_complementV02():
  ma = MultiAxis(())
  assert( list(ma.iter_complement('chicken')) == [()] )