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
|
# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE
from __future__ import annotations
import datetime
import numpy as np
import pytest # noqa: F401
import awkward as ak
to_list = ak.operations.to_list
def test_date_time():
numpy_array = np.array(
["2020-07-27T10:41:11", "2019-01-01", "2020-01-01"], "datetime64[s]"
)
array = ak.contents.NumpyArray(numpy_array)
assert str(array.form.type) == "datetime64[s]"
assert to_list(array) == [
np.datetime64("2020-07-27T10:41:11"),
np.datetime64("2019-01-01T00:00:00"),
np.datetime64("2020-01-01T00:00:00"),
]
for i in range(len(array)):
assert array[i] == numpy_array[i]
date_time = np.datetime64("2020-07-27T10:41:11.200000011", "us")
array1 = ak.contents.NumpyArray(
np.array(["2020-07-27T10:41:11.200000011"], "datetime64[us]")
)
assert np.datetime64(array1[0], "us") == date_time
assert to_list(ak.operations.from_iter(array1)) == [
np.datetime64("2020-07-27T10:41:11.200000")
]
def test_date_time_sort_argsort_unique():
numpy_array = np.array(
["2020-07-27T10:41:11", "2019-01-01", "2020-01-01"], "datetime64[s]"
)
array = ak.contents.NumpyArray(numpy_array)
assert to_list(ak.sort(array, highlevel=False)) == [
datetime.datetime(2019, 1, 1, 0, 0),
datetime.datetime(2020, 1, 1, 0, 0),
datetime.datetime(2020, 7, 27, 10, 41, 11),
]
assert to_list(ak.argsort(array, highlevel=False)) == [1, 2, 0]
assert ak._do.is_unique(array) is True
assert to_list(ak._do.unique(array)) == [
datetime.datetime(2019, 1, 1, 0, 0),
datetime.datetime(2020, 1, 1, 0, 0),
datetime.datetime(2020, 7, 27, 10, 41, 11),
]
def test_time_delta_sort_argsort_unique():
numpy_array = np.array(["41", "1", "20"], "timedelta64[D]")
array = ak.highlevel.Array(numpy_array).layout
assert str(array.form.type) == "timedelta64[D]"
assert to_list(array) == [
np.timedelta64("41", "D"),
np.timedelta64("1", "D"),
np.timedelta64("20", "D"),
]
assert to_list(ak.sort(array, highlevel=False)) == [
datetime.timedelta(days=1),
datetime.timedelta(days=20),
datetime.timedelta(days=41),
]
assert to_list(ak.argsort(array, highlevel=False)) == [1, 2, 0]
assert ak._do.is_unique(array) is True
assert to_list(ak._do.unique(array)) == [
datetime.timedelta(days=1),
datetime.timedelta(days=20),
datetime.timedelta(days=41),
]
|
