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())