def test_index_values_at_depth_b(self) -> None: idx1 = IndexGO(('a', 'b', 'c', 'd', 'e')) with self.assertRaises(RuntimeError): idx1.values_at_depth([3, 4]) post = idx1.values_at_depth([0]) self.assertEqual(post.tolist(), ['a', 'b', 'c', 'd', 'e'])
def test_index_values_at_depth_a(self) -> None: idx1 = IndexGO(('a', 'b', 'c')) self.assertEqual(idx1.values_at_depth(0).tolist(), ['a', 'b', 'c']) with self.assertRaises(RuntimeError): idx1.values_at_depth(1)