(a, b, omit=None, *, layers: bool)
| 36 | |
| 37 | |
| 38 | def assert_no_common_keys(a, b, omit=None, *, layers: bool) -> None: |
| 39 | dsk1 = a.__dask_graph__() |
| 40 | dsk2 = b.__dask_graph__() |
| 41 | |
| 42 | if omit is not None: |
| 43 | dsko = omit.__dask_graph__() |
| 44 | assert not (dsk1.keys() - dsko.keys()) & dsk2.keys() |
| 45 | assert not dsko.keys() - dsk1.keys() |
| 46 | assert not dsko.keys() - dsk2.keys() |
| 47 | if layers: |
| 48 | assert not (dsk1.layers.keys() - dsko.layers.keys()) & dsk2.layers.keys() |
| 49 | assert ( |
| 50 | not (dsk1.dependencies.keys() - dsko.dependencies.keys()) |
| 51 | & dsk2.dependencies.keys() |
| 52 | ) |
| 53 | assert not dsko.layers.keys() - dsk1.layers.keys() |
| 54 | assert not dsko.layers.keys() - dsk2.layers.keys() |
| 55 | assert not dsko.dependencies.keys() - dsk1.dependencies.keys() |
| 56 | assert not dsko.dependencies.keys() - dsk2.dependencies.keys() |
| 57 | |
| 58 | else: |
| 59 | assert not dsk1.keys() & dsk2.keys() |
| 60 | if layers: |
| 61 | assert not dsk1.layers.keys() & dsk2.layers.keys() |
| 62 | assert not dsk1.dependencies.keys() & dsk2.dependencies.keys() |
| 63 | |
| 64 | |
| 65 | def assert_did_not_materialize(cloned, orig): |
no test coverage detected