TST: clean up a test and compare return values instead of just making sure the indexing op does not fail