Add missing `comm` argument to LocalArray constructor.

Segfaults otherwise...
Refactor test_umath...

to make a client and context per-testclass rather than at the module level.