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(