Address comments following code review
Do not raise warning when the old code was working correctly.
test the non broadcast case
Test no broadcast on all dimensions