I suppose what I want to test here is that `A[Z1]` has zero total shape. Do you think I should change it to this?

    assert_equal(sum(A[Z1].shape), 0)