code: def test_load_data(generate_categorical_data):
      df = generate_categorical_data()
  
      features = load_data(df, df.columns)
  
      assert df.shape[0] == features.shape[0]
  
imports: import pandas
  import pytest
  from causalml.features import OneHotEncoder, LabelEncoder, load_data
label_desc: Use image node_id 2 for calling a global function with example usage: test_load_data(generate_categorical_data) without return types