use `np.testing.assert_array_equal` here