Added test suite for set_index