def test_loc_map_b(self) -> None: idx = Index(['a', 'b', 'c', 'd', 'e']) post1 = LocMap.loc_to_iloc( label_to_pos=idx._map, labels=idx._labels, positions=idx._positions, key=['b', 'd'], partial_selection=False, ) self.assertEqual(post1, [1, 3])
def test_loc_map_a(self) -> None: idx = Index(['a', 'b', 'c']) post1 = LocMap.loc_to_iloc( label_to_pos=idx._map, labels=idx._labels, positions=idx._positions, key='b', partial_selection=False, ) self.assertEqual(post1, 1) post2 = LocMap.loc_to_iloc( label_to_pos=idx._map, labels=idx._labels, positions=idx._positions, key=NULL_SLICE, partial_selection=False, ) self.assertEqual(post2, NULL_SLICE)
def test_loc_map_slice_c(self) -> None: dt64 = np.datetime64 idx = IndexDate.from_date_range('1985-01-01', '1985-01-08') post1 = LocMap.loc_to_iloc( label_to_pos=idx._map, labels=idx._labels, positions=idx._positions, key=slice(dt64('1985-01-01'), dt64('1985-01-04')), partial_selection=False, ) self.assertEqual(post1, slice(0, 4, None))
def test_loc_map_slice_b(self) -> None: dt64 = np.datetime64 idx = IndexDate.from_date_range('1985-01-01', '1985-01-08') with self.assertRaises(RuntimeError): post1 = LocMap.loc_to_iloc( label_to_pos=idx._map, labels=idx._labels, positions=idx._positions, key=slice(dt64('1985-01-01'), dt64('1985-01-04'), dt64('1985-01-04')), partial_selection=False, )