Skip to content

Commit c31b35c

Browse files
committed
Revert "Try separate check and assert_type"
This reverts commit 6be36cf.
1 parent 6be36cf commit c31b35c

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

tests/test_scalars.py

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1387,16 +1387,20 @@ def test_timestamp_cmp() -> None:
13871387
)
13881388
assert (eq_arr != ne_arr).all()
13891389

1390-
assert_type(ts == c_np_ndarray_dt64, np_ndarray_bool)
1391-
eq_arr = check(ts == c_np_ndarray_dt64, np.ndarray, np.bool_)
1392-
assert_type(ts != c_np_ndarray_dt64, np_ndarray_bool)
1393-
ne_arr = check(ts != c_np_ndarray_dt64, np.ndarray, np.bool_)
1390+
eq_arr = check(
1391+
assert_type(ts == c_np_ndarray_dt64, np_ndarray_bool), np.ndarray, np.bool_
1392+
)
1393+
ne_arr = check(
1394+
assert_type(ts != c_np_ndarray_dt64, np_ndarray_bool), np.ndarray, np.bool_
1395+
)
13941396
assert (eq_arr != ne_arr).all()
13951397
# TODO: the following should be 2D-arrays but it doesn't work in mypy
1396-
assert_type(ts == c_np_2darray_dt64, np_ndarray_bool)
1397-
eq_arr = check(ts == c_np_2darray_dt64, np_ndarray_bool)
1398-
assert_type(ts != c_np_2darray_dt64, np_ndarray_bool)
1399-
ne_arr = check(ts != c_np_2darray_dt64, np_ndarray_bool)
1398+
eq_arr = check(
1399+
assert_type(ts == c_np_2darray_dt64, np_ndarray_bool), np_ndarray_bool
1400+
)
1401+
ne_arr = check(
1402+
assert_type(ts != c_np_2darray_dt64, np_ndarray_bool), np_ndarray_bool
1403+
)
14001404
assert (eq_arr != ne_arr).all()
14011405

14021406
eq_s = check(

0 commit comments

Comments
 (0)