ENH: adding tests for loc