Add a correctness assertion for the groupby benchmark