DOC: added documention
Making tests
Added tests for trace, mean, axial and radial diffusivities