Make array size tests work with pytest on a notebook#9

nkalauni wants to merge 4 commits intoProjectPythia:mainProjectPythia/xbatcher-deep-learning:mainfrom

nkalauni:mainnkalauni/xbatcher-deep-learning:mainCopy head branch name to clipboard