SQLAlchemy: count number of executed statements

gistfile1.py

import sys

from sqlalchemy import create_engine, event
from sqlalchemy.orm import sessionmaker

import model

if sys.version_info >= (2, 7):
    import unittest as unittest2
else:
    import unittest2


class DBTest(unittest2.TestCase):

    def setUp(self):
        super(DBTest, self).setUp()
        self.engine = create_engine('sqlite://', echo=True)
        model.Base.metadata.create_all(self.engine)
        self.session_maker = sessionmaker(bind=self.engine)
        self.session = self.session_maker()
        self.nr_executed_statements = 0

        @event.listens_for(self.engine, 'after_execute')
        def after_execute(conn, clauseelement, multiparams, params, result):
            self.nr_executed_statements += 1

    def tearDown(self):
        super(DBTest, self).tearDown()
        self.session.close()