ML: Use testbook instead of pytest-notebook

ML: Use testbook instead of pytest-notebook #153