BUG: .sort() returns None and thus made the test moot. This revealed a bug with finding the axes that only contain nans.