def test_index_level_dtypes_all_a(self) -> None: level0 = IndexLevel(index=Index(('a', 'b')), targets=None) post = tuple(level0.dtypes_iter()) self.assertEqual(post[0], np.dtype('<U1')) self.assertEqual(len(post), 1) self.assertEqual(level0.depth, next(level0.depths()))
def test_index_level_dtypes_c(self) -> None: level0 = IndexLevel(index=Index(('a', 'b')), targets=None) post = tuple(level0.dtype_per_depth()) self.assertEqual(post[0], np.dtype('<U1')) self.assertEqual(len(post), 1)
def test_index_levels_equals_d(self) -> None: levels1 = IndexLevel(Index(('a', 'b', 'c')), targets=None) levels2 = IndexLevel(Index(('a', 'b', 'c')), targets=None) self.assertTrue(levels1.equals(levels2))
def test_index_levels_to_type_blocks_a(self) -> None: levels1 = IndexLevel(Index(()), targets=None, depth_reference=3) tb = levels1.to_type_blocks() # NOTE: this will be updated to (0, 0) with IndexLevel support for zero size self.assertEqual(tb.shape, (0, 3))
def test_index_level_loc_to_iloc_b(self) -> None: level0 = IndexLevel(index=Index(('a', 'b')), targets=None) with self.assertRaises(KeyError): level0.loc_to_iloc(HLoc['c', ])
def test_index_level_leaf_loc_to_iloc_b(self) -> None: level0 = IndexLevel(index=Index(('a', 'b')), targets=None) self.assertEqual(level0.leaf_loc_to_iloc(('b', )), 1) self.assertEqual(level0.leaf_loc_to_iloc(ILoc[1]), 1) #type: ignore
def test_index_level_contains_a(self) -> None: level0 = IndexLevel(index=Index(('a', 'b')), targets=None) self.assertFalse(('c', ) in level0) self.assertTrue(('a', ) in level0)
def test_index_level_index_types_b(self) -> None: level0 = IndexLevel(index=Index(('a', 'b')), targets=None) post = tuple(level0.index_types()) self.assertEqual(post[0], Index) self.assertEqual(len(post), 1)
def test_index_isin(self, labels: tp.Sequence[tp.Hashable]) -> None: index = Index(labels) self.assertTrue(index.isin((labels[0], ))[0])