Esempio n. 1
0
    def test_sel_datetime(self) -> None:
        index = PandasIndex(
            pd.to_datetime(["2000-01-01", "2001-01-01", "2002-01-01"]), "x")
        actual = index.sel({"x": "2001-01-01"})
        expected_dim_indexers = {"x": 1}
        assert actual.dim_indexers == expected_dim_indexers

        actual = index.sel({"x": index.to_pandas_index().to_numpy()[1]})
        assert actual.dim_indexers == expected_dim_indexers
Esempio n. 2
0
    def test_query_datetime(self):
        index = PandasIndex(
            pd.to_datetime(["2000-01-01", "2001-01-01", "2002-01-01"]), "x")
        actual = index.query({"x": "2001-01-01"})
        expected = (1, None)
        assert actual == expected

        actual = index.query({"x": index.to_pandas_index().to_numpy()[1]})
        assert actual == expected
Esempio n. 3
0
 def test_to_pandas_index(self) -> None:
     pd_idx = pd.Index([1, 2, 3], name="foo")
     index = PandasIndex(pd_idx, "x")
     assert index.to_pandas_index() is index.index
Esempio n. 4
0
 def to_pandas_index(self):
     pd_idx = pd.Index([1, 2, 3], name="foo")
     index = PandasIndex(pd_idx, "x")
     assert index.to_pandas_index() is pd_idx