In [1]:
from jupyterworkflow.data import get_fremont_data
import pandas as pd
def test_fremont_data():
data = get_fremont_data()
assert all(data.columns == ['west', 'east', 'Total'])
assert isinstance(data.index, pd.DatetimeIndex)
In [2]:
test_fremont_data()