[WIP] Rewrite docs tests into new code

Currently **WIP**. Will return back to it.