def _(dtype):
kwargs = {}
if PANDAS_VERSION >= "0.24.0rc1":
kwargs['dtype'] = dtype
return DecimalArray._from_sequence([Decimal('0'), Decimal('NaN')],
**kwargs)
def test_register_extension_type():
arr = DecimalArray._from_sequence([Decimal('1.0')] * 10)
ser = pd.Series(arr)
dser = dd.from_pandas(ser, 2)
assert_eq(ser, dser)
df = pd.DataFrame({"A": ser})
ddf = dd.from_pandas(df, 2)
assert_eq(df, ddf)
def data_for_twos():
return DecimalArray([decimal.Decimal(2) for _ in range(100)])
def data_missing():
return DecimalArray([decimal.Decimal("NaN"), decimal.Decimal(1)])
def data():
return DecimalArray(make_data())