_check_array should error on 0-dim also
fix handling of shapes when K=1
add some simple tests