1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143
|
# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE
from __future__ import annotations
import io
from datetime import datetime, timedelta
import numpy as np
import pytest
import awkward as ak
def test_precision():
stream = io.StringIO()
ak.Array([1.0, 2.3456789]).show(stream=stream, precision=1)
assert stream.getvalue() == "[1,\n 2]\n"
stream.seek(0)
ak.Array([1.0, 2.3456789]).show(stream=stream, precision=7)
assert stream.getvalue() == "[1,\n 2.345679]\n"
def test_float():
stream = io.StringIO()
ak.Array([1.0, 2.3456789]).show(
stream=stream, formatter={"float": "<FLOAT {}>".format}
)
assert stream.getvalue() == "[<FLOAT 1.0>,\n <FLOAT 2.3456789>]\n"
stream.seek(0)
ak.Array([1.0, 2.3456789]).show(
stream=stream, formatter={"float_kind": "<FLOAT {}>".format}
)
assert stream.getvalue() == "[<FLOAT 1.0>,\n <FLOAT 2.3456789>]\n"
@pytest.mark.skipif(
not hasattr(np, "float128"), reason="Only for 128-float supporting platforms"
)
def test_longfloat():
stream = io.StringIO()
ak.values_astype([1.0, 2.3456789], "float128").show(
stream=stream, formatter={"longfloat": "<FLOAT {}>".format}
)
assert stream.getvalue() == "[<FLOAT 1.0>,\n <FLOAT 2.3456789>]\n"
stream.seek(0)
ak.values_astype([1.0, 2.3456789], "float128").show(
stream=stream, formatter={"float_kind": "<FLOAT {}>".format}
)
assert stream.getvalue() == "[<FLOAT 1.0>,\n <FLOAT 2.3456789>]\n"
def test_complex():
stream = io.StringIO()
ak.Array([1.0j, 2.3456789j + 2]).show(
stream=stream, formatter={"complexfloat": "<COMPLEX {}>".format}
)
assert stream.getvalue() == "[<COMPLEX 1j>,\n <COMPLEX (2+2.3456789j)>]\n"
stream.seek(0)
ak.Array([1.0j, 2.3456789j + 2]).show(
stream=stream, formatter={"complex_kind": "<COMPLEX {}>".format}
)
assert stream.getvalue() == "[<COMPLEX 1j>,\n <COMPLEX (2+2.3456789j)>]\n"
@pytest.mark.skipif(
not hasattr(np, "complex256"), reason="Only for 256-complex supporting platforms"
)
def test_longcomplex():
stream = io.StringIO()
ak.values_astype([1.0j, 2.3456789j + 2], "complex256").show(
stream=stream, formatter={"longcomplexfloat": "<COMPLEX {}>".format}
)
assert stream.getvalue() == "[<COMPLEX 1j>,\n <COMPLEX (2+2.3456789j)>]\n"
stream.seek(0)
ak.values_astype([1.0j, 2.3456789j + 2], "complex256").show(
stream=stream, formatter={"complex_kind": "<COMPLEX {}>".format}
)
assert stream.getvalue() == "[<COMPLEX 1j>,\n <COMPLEX (2+2.3456789j)>]\n"
def test_int():
stream = io.StringIO()
ak.Array([1, 2]).show(stream=stream, formatter={"int": "<INT {}>".format})
assert stream.getvalue() == "[<INT 1>,\n <INT 2>]\n"
stream.seek(0)
ak.Array([1, 2]).show(stream=stream, formatter={"int_kind": "<INT {}>".format})
assert stream.getvalue() == "[<INT 1>,\n <INT 2>]\n"
def test_bool():
stream = io.StringIO()
ak.Array([True, False]).show(stream=stream, formatter={"bool": "<BOOL {}>".format})
assert stream.getvalue() == "[<BOOL True>,\n <BOOL False>]\n"
def test_str():
stream = io.StringIO()
ak.Array(["hello", "world"]).show(
stream=stream, formatter={"str": "<STRING {!r}>".format}
)
assert stream.getvalue() == "[<STRING 'hello'>,\n <STRING 'world'>]\n"
stream.seek(0)
ak.Array(["hello", "world"]).show(
stream=stream, formatter={"str_kind": "<STRING {!r}>".format}
)
assert stream.getvalue() == "[<STRING 'hello'>,\n <STRING 'world'>]\n"
def test_bytes():
stream = io.StringIO()
ak.Array([b"hello", b"world"]).show(
stream=stream, formatter={"bytes": "<BYTES {!r}>".format}
)
assert stream.getvalue() == "[<BYTES b'hello'>,\n <BYTES b'world'>]\n"
stream.seek(0)
ak.Array([b"hello", b"world"]).show(
stream=stream, formatter={"str_kind": "<STRING {!r}>".format}
)
assert stream.getvalue() == "[<STRING b'hello'>,\n <STRING b'world'>]\n"
def test_datetime64():
stream = io.StringIO()
ak.Array([datetime(year=2023, month=1, day=1, hour=12, minute=1, second=30)]).show(
stream=stream, formatter={"datetime": "<DT {}>".format}
)
assert stream.getvalue() == "[<DT 2023-01-01T12:01:30.000000>]\n"
def test_timedelta64():
stream = io.StringIO()
ak.Array([timedelta(days=1, hours=12, minutes=1, seconds=30)]).show(
stream=stream, formatter={"datetime": "<TD {}>".format}
)
assert stream.getvalue() == "[129690000000 microseconds]\n"
|