import jupyterworkflow.data as jwd
import pandas as pd
import numpy as np

def test_fremont_data():
    data = jwd.get_fremont_data()
    assert all(data.columns == [u'East', u'West', u'Total'])
    assert isinstance(data.index, pd.DatetimeIndex)
    assert len(np.unique(data.index.time)) == 24

import jupyterworkflow.data as jwd
import pandas as pd
import numpy as np

def test_fremont_data():
    data = jwd.get_fremont_data()
    assert all(data.columns == [u'East', u'West', u'Total'])
    assert isinstance(data.index, pd.DatetimeIndex)
    assert len(np.unique(data.index.time)) == 24
