Improved coverage, cleaned up code, docstrings and tests.

Used list instead of dict for _TensorManager._comm