Python get_sample_data_frame Examples

Example #1

File: io_core_steps.py Project: savadev/kedro

def create_memory_data_frame(context):
    context.test_data_frame = get_sample_data_frame()
    context.memory_data_set = MemoryDataSet(data=context.test_data_frame)

Example #2

File: io_core_steps.py Project: savadev/kedro

def save_and_load_content(context):
    data_frame = get_sample_data_frame()
    context.csv_data_set.save(data_frame)
    context.reloaded_df = context.csv_data_set.load()

Example #3

File: io_core_steps.py Project: savadev/kedro

def reloaded_content_is_returned(context):
    data_frame = get_sample_data_frame()
    assert context.reloaded_df.equals(data_frame)

Example #4

File: io_core_steps.py Project: savadev/kedro

def save_data_frame(context):
    data_frame = get_sample_data_frame()
    try:
        context.csv_data_set.save(data_frame)
    except DataSetError:
        context.data_set_error_raised = True