TST: Add test coverage for agg methods