Make test coverage 100%#349

neuralsorcerer wants to merge 2 commits intofacebookresearch:mainfacebookresearch/balance:mainfrom

neuralsorcerer:testneuralsorcerer/balance:testCopy head branch name to clipboard