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