Add unit test to catch non-integer output shape bug