File: test_1765_add_ioanas_test_of_to_arraylib.py

package info (click to toggle)
python-awkward 2.8.9-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 24,932 kB
  • sloc: python: 178,875; cpp: 33,828; sh: 432; makefile: 21; javascript: 8
file content (35 lines) | stat: -rw-r--r-- 994 bytes parent folder | download
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
# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE

from __future__ import annotations

import numpy as np
import pytest  # noqa: F401

import awkward as ak


def test():
    aa = ak.contents.NumpyArray(np.frombuffer(b"hellothere", "u1"))
    b = aa.to_backend_array(allow_missing=False)
    assert b.tolist() == [104, 101, 108, 108, 111, 116, 104, 101, 114, 101]
    assert b.dtype == np.dtype(np.uint8)

    c = ak.contents.NumpyArray(np.array([0, 1577836800], dtype="datetime64[s]"))
    assert [
        d.isoformat() for d in c.to_backend_array(allow_missing=False).tolist()
    ] == [
        "1970-01-01T00:00:00",
        "2020-01-01T00:00:00",
    ]

    recordarray = ak.contents.RecordArray(
        [ak.contents.NumpyArray(np.array([1, 2, 3, 4, 5], dtype=np.int64))],
        fields=["one"],
    )
    assert recordarray.to_backend_array(allow_missing=False).tolist() == [
        (1,),
        (2,),
        (3,),
        (4,),
        (5,),
    ]