def test_direction_unit_length(eval_buffer): """ All directions must be unit length """ for v in numpy.nditer(eval_buffer.array): assert v["x"]**2 + v["y"]**2 + v["z"]**2 == pytest.approx(1)