Remove test import

This removes the namespace import of the test subpackage, which explicitly imports mpi4py. The pshmem package can now be imported without triggering an mpi4py import. mpi4py will only be used if a communicator is passed which is not None.