| 189 | |
| 190 | |
| 191 | def test_arange(): |
| 192 | darr = da.arange(77, chunks=13) |
| 193 | nparr = np.arange(77) |
| 194 | assert_eq(darr, nparr) |
| 195 | |
| 196 | darr = da.arange(2, 13, chunks=5) |
| 197 | nparr = np.arange(2, 13) |
| 198 | assert_eq(darr, nparr) |
| 199 | |
| 200 | darr = da.arange(4, 21, 9, chunks=13) |
| 201 | nparr = np.arange(4, 21, 9) |
| 202 | assert_eq(darr, nparr) |
| 203 | |
| 204 | # negative steps |
| 205 | darr = da.arange(53, 5, -3, chunks=5) |
| 206 | nparr = np.arange(53, 5, -3) |
| 207 | assert_eq(darr, nparr) |
| 208 | |
| 209 | darr = da.arange(77, chunks=13, dtype=float) |
| 210 | nparr = np.arange(77, dtype=float) |
| 211 | assert_eq(darr, nparr) |
| 212 | |
| 213 | darr = da.arange(2, 13, chunks=5, dtype=int) |
| 214 | nparr = np.arange(2, 13, dtype=int) |
| 215 | assert_eq(darr, nparr) |
| 216 | assert sorted(da.arange(2, 13, chunks=5).dask) == sorted( |
| 217 | da.arange(2, 13, chunks=5).dask |
| 218 | ) |
| 219 | assert sorted(da.arange(77, chunks=13, dtype=float).dask) == sorted( |
| 220 | da.arange(77, chunks=13, dtype=float).dask |
| 221 | ) |
| 222 | |
| 223 | # 0 size output |
| 224 | darr = da.arange(0, 1, -0.5, chunks=20) |
| 225 | nparr = np.arange(0, 1, -0.5) |
| 226 | assert_eq(darr, nparr) |
| 227 | |
| 228 | darr = da.arange(0, -1, 0.5, chunks=20) |
| 229 | nparr = np.arange(0, -1, 0.5) |
| 230 | assert_eq(darr, nparr) |
| 231 | |
| 232 | # stop and/or step as kwargs |
| 233 | darr = da.arange(stop=10) |
| 234 | nparr = np.arange(stop=10) |
| 235 | assert_eq(darr, nparr) |
| 236 | |
| 237 | darr = da.arange(10, step=2) |
| 238 | nparr = np.arange(10, step=2) |
| 239 | assert_eq(darr, nparr) |
| 240 | |
| 241 | darr = da.arange(stop=10, step=2) |
| 242 | nparr = np.arange(stop=10, step=2) |
| 243 | assert_eq(darr, nparr) |
| 244 | |
| 245 | darr = da.arange(3, stop=10, step=2) |
| 246 | nparr = np.arange(3, stop=10, step=2) |
| 247 | assert_eq(darr, nparr) |
| 248 | |