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,
)