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
|
# This file is part of Hypothesis, which may be found at
# https://github.com/HypothesisWorks/hypothesis/
#
# Copyright the Hypothesis Authors.
# Individual contributors are listed in AUTHORS.rst and the git log.
#
# This Source Code Form is subject to the terms of the Mozilla Public License,
# v. 2.0. If a copy of the MPL was not distributed with this file, You can
# obtain one at https://mozilla.org/MPL/2.0/.
from datetime import datetime
import pandas as pd
from hypothesis import given, strategies as st
from hypothesis.extra import pandas as pdst
from tests.common.arguments import argument_validation_test, e
from tests.common.utils import checks_deprecated_behaviour
BAD_ARGS = [
e(pdst.data_frames),
e(pdst.data_frames, pdst.columns(1, dtype="not a dtype")),
e(pdst.data_frames, pdst.columns(1, elements="not a strategy")),
e(pdst.data_frames, pdst.columns([[]])),
e(pdst.data_frames, [], index=[]),
e(pdst.data_frames, [], rows=st.fixed_dictionaries({"A": st.just(1)})),
e(pdst.data_frames, pdst.columns(1)),
e(pdst.data_frames, pdst.columns(1, dtype=float, fill=1)),
e(pdst.data_frames, pdst.columns(1, dtype=float, elements=1)),
e(pdst.data_frames, pdst.columns(1, fill=1, dtype=float)),
e(pdst.data_frames, pdst.columns(["A", "A"], dtype=float)),
e(pdst.data_frames, pdst.columns(1, elements=st.none(), dtype=int)),
e(pdst.data_frames, 1),
e(pdst.data_frames, [1]),
e(pdst.data_frames, pdst.columns(1, dtype="category")),
e(
pdst.data_frames,
pdst.columns(["A"], dtype=bool),
rows=st.tuples(st.booleans(), st.booleans()),
),
e(
pdst.data_frames,
pdst.columns(1, elements=st.booleans()),
rows=st.tuples(st.booleans()),
),
e(pdst.data_frames, rows=st.integers(), index=pdst.range_indexes(0, 0)),
e(pdst.data_frames, rows=st.integers(), index=pdst.range_indexes(1, 1)),
e(pdst.data_frames, pdst.columns(1, dtype=int), rows=st.integers()),
e(
pdst.data_frames,
columns=pdst.columns(["a", "b"], dtype=str, elements=st.text()),
rows=st.just({"a": "x"}),
index=pdst.indexes(dtype=int, min_size=1),
),
e(
pdst.data_frames,
columns=pdst.columns(["a", "b"], dtype=str, elements=st.text()),
rows=st.just(["x"]),
index=pdst.indexes(dtype=int, min_size=1),
),
e(pdst.indexes),
e(pdst.indexes, dtype="category"),
e(pdst.indexes, dtype="not a dtype"),
e(pdst.indexes, elements="not a strategy"),
e(pdst.indexes, elements=st.text(), dtype=float),
e(pdst.indexes, elements=st.none(), dtype=int),
e(pdst.indexes, elements=st.integers(0, 10), dtype=st.sampled_from([int, float])),
e(pdst.indexes, dtype=int, max_size=0, min_size=1),
e(pdst.indexes, dtype=int, unique="true"),
e(pdst.indexes, dtype=int, min_size="0"),
e(pdst.indexes, dtype=int, max_size="1"),
e(pdst.range_indexes, 1, 0),
e(pdst.range_indexes, min_size="0"),
e(pdst.range_indexes, max_size="1"),
e(pdst.range_indexes, name=""),
e(pdst.series),
e(pdst.series, dtype="not a dtype"),
e(pdst.series, elements="not a strategy"),
e(pdst.series, elements=st.none(), dtype=int),
e(pdst.series, dtype="category"),
e(pdst.series, index="not a strategy"),
]
test_raise_invalid_argument = argument_validation_test(BAD_ARGS)
lo, hi = pd.Timestamp(2017, 1, 1), pd.Timestamp(2084, 12, 21)
@given(st.datetimes(min_value=lo, max_value=hi))
def test_timestamp_as_datetime_bounds(dt):
# Would have caught https://github.com/HypothesisWorks/hypothesis/issues/2406
assert isinstance(dt, datetime)
assert lo <= dt <= hi
assert not isinstance(dt, pd.Timestamp)
@checks_deprecated_behaviour
def test_confusing_object_dtype_aliases():
pdst.series(elements=st.tuples(st.integers()), dtype=tuple).example()
|
