def test_rotation(self): phase = np.zeros((50, 50)) phase0 = pst._find_phase(phase, rotation=90) assert (phase0 == np.pi / 2).all() phase1 = pst._find_phase(phase, rotation=45) assert (phase1 == np.pi / 4).all() phase2 = pst._find_phase(phase, rotation=180) assert (phase2 == np.pi).all() phase3 = pst._find_phase(phase, rotation=360) assert (phase3 == 0).all() phase4 = pst._find_phase(phase, rotation=-90) assert (phase4 == 3 * np.pi / 2).all()
def test_max_phase(self): phase = (np.ones((50, 50)) * np.pi * 0.5) + np.pi phase0 = pst._find_phase(phase, max_phase=np.pi) assert (phase0 == np.pi / 2).all()
def test_simple(self): phase = np.zeros((50, 50)) phase0 = pst._find_phase(phase) assert (phase0 == 0.0).all()