def test_compute_omega(): rx, x = compute_omega(traj, opt=True) ry, y = compute_omega(traj, opt=False) eq(rx, ry) eq(x, y)