Explicitly check dtype in assert_array_equal. Adjust a few test cases that had the wrong dtype expectation (np.arange usually uses a different dtype from np.zeros).

Explicitly check dtype in assert_array_equal. Adjust a few test cases that had the wrong dtype expectation (np.arange usually uses a different dtype from np.zeros). #17472