def METHOD_NAME():
    A = np.random.randint(1, 10, size=(10, 10, 10), dtype=np.int32)
    B = np.empty((10, ), dtype=np.int32)
    C = np.zeros((10, ), dtype=np.int32)
    W = np.random.randint(2, size=(10, 10, 10), dtype=np.bool_)
    with dace.config.set_temporary('frontend', 'avoid_wcr', value=True):
        test_sdfg = augassign_wcr2.to_sdfg(simplify=False)
    wcr_count = 0
    for sdfg in test_sdfg.sdfg_list:
        for state in sdfg.nodes():
            for edge in state.edges():
                if edge.data.wcr:
                    wcr_count += 1
    assert (wcr_count == 2)
    count = test_sdfg(A=A, B=B, W=W)
    C = np.add.reduce(A, axis=(1, 2), where=W)
    assert (count[0] == np.count_nonzero(W))
    assert (np.array_equal(B, C))