def test_observed_values_plain(dtype, value): ind = ExplicitSecondaryIndex(column="col", dtype=dtype, index_dct={value: ["part_label"]}) observed = ind.observed_values() assert len(observed) == 1 assert list(observed) == [value]
def test_observed_values_date_as_object(date_as_object): value = datetime.date(2020, 1, 1) ind = ExplicitSecondaryIndex(column="col", dtype=pa.date32(), index_dct={value: ["part_label"]}) observed = ind.observed_values(date_as_object=date_as_object) if date_as_object: expected = value else: expected = pd.Timestamp(value).to_datetime64() assert len(observed) == 1 assert observed[0] == expected