예제 #1
0
 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()))
예제 #2
0
 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)
예제 #3
0
    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))
예제 #4
0
    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))
예제 #5
0
 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', ])
예제 #6
0
    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
예제 #7
0
 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)
예제 #8
0
 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)
예제 #9
0
 def test_index_isin(self, labels: tp.Sequence[tp.Hashable]) -> None:
     index = Index(labels)
     self.assertTrue(index.isin((labels[0], ))[0])