(shape)
| 1614 | |
| 1615 | @pytest.mark.parametrize("shape", [(1,), (1, 1)]) |
| 1616 | def test_squeeze_1d_array(shape): |
| 1617 | a = np.full(shape=shape, fill_value=2) |
| 1618 | a_s = np.squeeze(a) |
| 1619 | d = da.from_array(a, chunks=(1)) |
| 1620 | d_s = da.squeeze(d) |
| 1621 | assert isinstance(d_s, da.Array) |
| 1622 | assert isinstance(d_s.compute(), np.ndarray) |
| 1623 | assert_eq(d_s, a_s) |
| 1624 | |
| 1625 | |
| 1626 | def test_vstack(): |