()
| 394 | |
| 395 | |
| 396 | def test_from_array(): |
| 397 | arr = np.random.randint(1, 100, (100,)) |
| 398 | assert_eq(from_array(arr, chunksize=5), pd.Series(arr)) |
| 399 | assert_eq(from_array(arr, chunksize=5, columns="a"), pd.Series(arr, name="a")) |
| 400 | columns = ["a", "b", "c", "d"] |
| 401 | arr = np.random.randint(1, 100, (100, 4)) |
| 402 | assert_eq( |
| 403 | from_array(arr, chunksize=5, columns=columns), |
| 404 | pd.DataFrame(arr, columns=columns), |
| 405 | ) |
| 406 | assert_eq( |
| 407 | from_array(arr, chunksize=5, columns=columns)["a"], |
| 408 | pd.DataFrame(arr, columns=columns)["a"], |
| 409 | ) |
| 410 | assert_eq( |
| 411 | from_array(arr, chunksize=5, columns=columns)[["a"]], |
| 412 | pd.DataFrame(arr, columns=columns)[["a"]], |
| 413 | ) |
| 414 | q = from_array(arr, chunksize=5, columns=columns)[["a"]].simplify() |
| 415 | for expr in q.walk(): |
| 416 | if isinstance(expr, FromArray): |
| 417 | assert expr.columns == ["a"] |
| 418 | |
| 419 | |
| 420 | @pytest.mark.parametrize("dtype", [object, str]) |
nothing calls this directly
no test coverage detected