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
