Add test for optional size argument for ndindex