bug(db): matchit_cem_summary_statistics not working with grouping_attribute #5

SELECT bin_equal_width(
  'flights_weather_demo_airports',
  ARRAY['vism', 'wspdm'],
  ARRAY[10, 9],
  'flights_weather_demo_binned'
);

SELECT matchit_cem(
  'flights_weather_demo_binned',
  'fid',
  'lowpressure',
  ARRAY['vism_ew_binned_10', 'wspdm_ew_binned_9', 'thunder', 'fog', 'rain', 'snow'],
  'flights_weather_demo_matched'
);

SELECT matchit_cem_summary_statistics(
  'flights_weather_demo_binned',
  'flights_weather_demo_matched',
  'lowpressure',
  'depdel15',
  'airport',
  ARRAY['vism', 'wspdm', 'thunder', 'fog', 'rain', 'snow'],
  ARRAY['vism', 'wspdm'],
  ARRAY['vism_ew_binned_10', 'wspdm_ew_binned_9']
);

CREATE TABLE flights_weather_demo_matched_ate_cem AS
WITH Blocks AS 
    (SELECT avg(depdel15) AS avg_outcome,
         subclass_id,
         lowpressure,
         airport
    FROM flights_weather_demo_matched
    GROUP BY  subclass_id, lowpressure, airport), Weights AS 
    (SELECT cast(count(*) AS NUMERIC) / 8450 AS block_weight,
         subclass_id
    FROM flights_weather_demo_matched
    GROUP BY  subclass_id, airport)
SELECT Blocks.lowpressure AS treatment,
         sum(avg_outcome * block_weight) AS weighted_avg_outcome,
         airport
FROM Blocks, Weights
WHERE Blocks.subclass_id = Weights.subclass_id
GROUP BY  Blocks.lowpressure,
         airport;
         
SELECT json_agg(
  json_build_object(
    'treatment', treatment, 
    'weighted_avg_outcome', weighted_avg_outcome, 
    'grouping_attribute', airport
  )
)
FROM flights_weather_demo_matched_ate_cem
GROUP BY  airport;

CREATE OR REPLACE FUNCTION get_json_ate(
  table_name TEXT,
  treatment TEXT,
  grouping_attribute TEXT
) RETURNS JSONB AS $func$
DECLARE
  command_string TEXT;
  result_arr JSONB;
BEGIN
  command_string := 'SELECT json_agg(json_build_object('
    || '''treatment'', treatment, '
    || '''weighted_avg_outcome'', weighted_avg_outcome';

  IF grouping_attribute IS NOT NULL THEN
    command_string := command_string || ', ''grouping_attribute'', ' || quote_ident(grouping_attribute);
  END IF;

  command_string := command_string || ')) FROM ' || quote_ident(table_name);

  IF grouping_attribute IS NOT NULL THEN
    command_string := command_string || ' GROUP BY ' || quote_ident(grouping_attribute);
  END IF;

  RAISE NOTICE '%', command_string;
  EXECUTE command_string INTO result_arr;

  RETURN result_arr;
END;
$func$ LANGUAGE plpgsql;