old_contents: #!/usr/bin/env python
  
  
  class TestGenSql:
  
      def test_gen_drop_statement(self):
          pass
  
      def test_create_statement(self):
          pass
  
new_contents: #!/usr/bin/env python
  import sys
  from io import StringIO
  
  from pg_bawler import gen_sql
  
  
  def test_simple_main(monkeypatch):
      stdout = StringIO()
      monkeypatch.setattr(sys, 'stdout', stdout)
  
      class Args:
          tablename = 'foo'
  
      gen_sql.main(*[Args.tablename])
      sql = stdout.getvalue()
  
      assert gen_sql.TRIGGER_FN_FMT.format(args=Args) in sql
      assert gen_sql.TRIGGER_FN_FMT.format(args=Args) in sql
  
subject: Add simple test for sql full sql generation with only tablename