def chunkedStops(acs):
stops = Stop(stop_filepath="data/stop_data/stops_test_no_officer_id.csv",
chunk=True,
acs=acs,
chunksize=10)
return stops.create_summary()
def test_drop_rows_with_no_driver_race_data():
stops = Stop(stop_filepath="data/stop_data/stops_test_no_driver_race.csv")
stops.create_summary()
assert len(stops.df) == 0
def stops_summary():
stops = Stop(stop_filepath='data/stop_data/stops_test.csv')
stops.create_summary()
return stops
def test_files_with_no_officer_id_loads():
stops = Stop(stop_filepath="data/stop_data/stops_test_no_officer_id.csv")
stops.create_summary()
assert len(stops.df) > 5