def METHOD_NAME(self, target_schema, table, count_in_source):
    """
    Check if row counts are equal in source and target
    """
    self.assertEqual(
        count_in_source,
        self.e2e_env.run_query_target_snowflake(
            f'select count(_id) from {target_schema}.{table}'
        )[0][0],
    )