Add more tests for collapse handler#809

fehiepsi wants to merge 15 commits into

pyro-ppl:masterpyro-ppl/numpyro:masterfrom

fehiepsi:collapse_platefehiepsi/numpyro:collapse_plateCopy head branch name to clipboard