Tests: replace Axes.CustomType use

The nested class Axes.CustomType has been replaced by DefaultAxisType.
Updated use cases in tests as necessary.