()
| 361 | |
| 362 | |
| 363 | def test_dot(): |
| 364 | import dask.array as da |
| 365 | |
| 366 | s1 = pd.Series([1, 2, 3, 4]) |
| 367 | s2 = pd.Series([4, 5, 6, 6]) |
| 368 | df = pd.DataFrame({"one": s1, "two": s2}) |
| 369 | |
| 370 | dask_s1 = from_pandas(s1, npartitions=1) |
| 371 | dask_df = from_pandas(df, npartitions=1) |
| 372 | dask_s2 = from_pandas(s2, npartitions=1) |
| 373 | |
| 374 | assert_eq(s1.dot(s2), dask_s1.dot(dask_s2)) |
| 375 | assert_eq(s1.dot(df), dask_s1.dot(dask_df)) |
| 376 | |
| 377 | # With partitions |
| 378 | partitioned_s1 = from_pandas(s1, npartitions=2) |
| 379 | partitioned_df = from_pandas(df, npartitions=2) |
| 380 | partitioned_s2 = from_pandas(s2, npartitions=2) |
| 381 | |
| 382 | assert_eq(s1.dot(s2), partitioned_s1.dot(partitioned_s2)) |
| 383 | assert_eq(s1.dot(df), partitioned_s1.dot(partitioned_df)) |
| 384 | |
| 385 | # Test passing meta kwarg |
| 386 | res = dask_s1.dot(dask_df, meta=pd.Series([1], name="test_series")).compute() |
| 387 | assert res.name == "test_series" |
| 388 | |
| 389 | # Test validation of second operand |
| 390 | with pytest.raises(TypeError): |
| 391 | dask_s1.dot(da.array([1, 2, 3, 4])) |
| 392 | |
| 393 | |
| 394 | def test_pipe(): |
nothing calls this directly
no test coverage detected