address comments

Author: chelsea-lin

tests/unit/core/compile/sqlglot/aggregations/test_op_registration.py

@@ -32,7 +32,7 @@ def test_func(op: agg_ops.SizeOp, input: sge.Expression) -> sge.Expression:
     assert reg[agg_ops.SizeOp.name](op, input) == test_func(op, input)
 
 
-def test_register_function_first_argument_is_not_scalar_op_raise_error():
+def test_register_function_first_argument_is_not_agg_op_raise_error():
     reg = op_registration.OpRegistration()
 
     @reg.register(agg_ops.SizeOp)

tests/unit/core/compile/sqlglot/aggregations/test_unary_compiler.py

@@ -40,10 +40,12 @@ def _apply_unary_op(obj: bpd.DataFrame, op: agg_ops.UnaryWindowOp, arg: str) ->
 def test_size(scalar_types_df: bpd.DataFrame, snapshot):
     bf_df = scalar_types_df[["string_col"]]
     sql = _apply_unary_op(bf_df, agg_ops.SizeUnaryOp(), "string_col")
+
     snapshot.assert_match(sql, "out.sql")
 
 
 def test_sum(scalar_types_df: bpd.DataFrame, snapshot):
     bf_df = scalar_types_df[["int64_col"]]
     sql = _apply_unary_op(bf_df, agg_ops.SumOp(), "int64_col")
+
     snapshot.assert_match(sql, "out.sql")