Fix dismax tests.

I also added some of the edismax specific tests from #371.