cast axis arguments as ints
test and fix issues with force_floatX