from time import time as TIMENOW
import os.path as op

NBDIR = '../examples/Notebooks/'

def run_nb(nb):
    """ Run a notebook in NBDIR; adds .ipynb extension if none found. """
    if not op.splitext(nb)[1]:
        nb = '%s.ipynb' % nb
    nbpath = op.join(NBDIR, nb)
    %run $nbpath
    
run_nb('NBCONFIG.ipynb')

%%capture CAPTURED
ts = TIMENOW()
NB = 'basictests.ipynb'
run_nb(NB)
tdiff = TIMENOW() - ts
'COMPLETED %s: %.2f sec' % (NB, tdiff)

%%capture CAPTURED
ts = TIMENOW()
NB = 'shapes.ipynb'
run_nb(NB)
tdiff = TIMENOW() - ts
'COMPLETED %s: %.2f sec' % (NB, tdiff)

%%capture CAPTURED
ts = TIMENOW()
NB = 'multi_tutorial.ipynb'
run_nb(NB)
tdiff = TIMENOW() - ts
'COMPLETED %s: %.2f sec' % (NB, tdiff)

%%capture CAPTURED
ts = TIMENOW()
NB = 'grids.ipynb'
run_nb(NB)
tdiff = TIMENOW() - ts
'COMPLETED %s: %.2f sec' % (NB, tdiff)

%%capture CAPTURED
ts = TIMENOW()
NB = 'Analyze_Particles.ipynb'
run_nb(NB)
tdiff = TIMENOW() - ts
'COMPLETED %s: %.2f sec' % (NB, tdiff)

%%capture CAPTURED
ts = TIMENOW()
NB = 'groups_of_labels.ipynb'
run_nb(NB)
tdiff = TIMENOW() - ts
'COMPLETED %s: %.2f sec' % (NB, tdiff)

%%capture CAPTURED
ts = TIMENOW()
NB = 'making_noise.ipynb'
run_nb(NB)
tdiff = TIMENOW() - ts
'COMPLETED %s: %.2f sec' % (NB, tdiff)

%%capture CAPTURED
ts = TIMENOW()
NB = 'JORS_data.ipynb'
run_nb(NB)
tdiff = TIMENOW() - ts
'COMPLETED %s: %.2f sec' % (NB, tdiff)