Add test for GetRangeByPartitionKey