flake8
use notebook_test directly
reorganizing tests