Try separate check and assert_type

Author: hamdanal

tests/test_scalars.py

@@ -1387,20 +1387,16 @@ def test_timestamp_cmp() -> None:
  )
  assert (eq_arr != ne_arr).all()
 
- eq_arr = check(
- assert_type(ts == c_np_ndarray_dt64, np_ndarray_bool), np.ndarray, np.bool_
- )
- ne_arr = check(
- assert_type(ts != c_np_ndarray_dt64, np_ndarray_bool), np.ndarray, np.bool_
- )
+ assert_type(ts == c_np_ndarray_dt64, np_ndarray_bool)
+ eq_arr = check(ts == c_np_ndarray_dt64, np.ndarray, np.bool_)
+ assert_type(ts != c_np_ndarray_dt64, np_ndarray_bool)
+ ne_arr = check(ts != c_np_ndarray_dt64, np.ndarray, np.bool_)
  assert (eq_arr != ne_arr).all()
  # TODO: the following should be 2D-arrays but it doesn't work in mypy
- eq_arr = check(
- assert_type(ts == c_np_2darray_dt64, np_ndarray_bool), np_ndarray_bool
- )
- ne_arr = check(
- assert_type(ts != c_np_2darray_dt64, np_ndarray_bool), np_ndarray_bool
- )
+ assert_type(ts == c_np_2darray_dt64, np_ndarray_bool)
+ eq_arr = check(ts == c_np_2darray_dt64, np_ndarray_bool)
+ assert_type(ts != c_np_2darray_dt64, np_ndarray_bool)
+ ne_arr = check(ts != c_np_2darray_dt64, np_ndarray_bool)
  assert (eq_arr != ne_arr).all()
 
  eq_s = check(