Add a test geometry for Axi and 2D
[ci skip]