[IMP] Improve parsing of unit tests
[IMP] Improve Unit test parsing (class docstring, method name, method
docstring)