(inline_array)
| 105 | |
| 106 | @pytest.mark.parametrize("inline_array", [True, False]) |
| 107 | def test_graph_from_arraylike(inline_array): |
| 108 | d = 2 |
| 109 | chunk = (2, 3) |
| 110 | shape = tuple(d * n for n in chunk) |
| 111 | arr = np.ones(shape) |
| 112 | |
| 113 | dsk = graph_from_arraylike( |
| 114 | arr, chunk, shape=shape, name="X", inline_array=inline_array |
| 115 | ) |
| 116 | |
| 117 | assert isinstance(dsk, HighLevelGraph) |
| 118 | if inline_array: |
| 119 | assert len(dsk.layers) == 1 |
| 120 | assert isinstance(hlg_layer_topological(dsk, 0), Blockwise) |
| 121 | else: |
| 122 | assert len(dsk.layers) == 2 |
| 123 | assert isinstance(hlg_layer_topological(dsk, 0), MaterializedLayer) |
| 124 | assert isinstance(hlg_layer_topological(dsk, 1), Blockwise) |
| 125 | dsk = dict(dsk) |
| 126 | |
| 127 | # Somewhat odd membership check to avoid numpy elemwise __in__ overload |
| 128 | assert any(arr is v for v in dsk.values()) is not inline_array |
| 129 | |
| 130 | |
| 131 | def test_top(): |
nothing calls this directly
no test coverage detected