Merge pull request #92 from jseabold/fix-integration-tests

Fix integration tests.
sqlalchemy-redshift · Aug 24, 2016 · 7691341 · 7691341
2 parents 471ce46 + 6a04627
commit 7691341
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/test_postfetch_lastrowid.py b/tests/test_postfetch_lastrowid.py
@@ -3,8 +3,7 @@
 
 def test_postfetch_lastrowid(redshift_session):
     session = redshift_session
-    for i in range(5):
-        postfetchExample = models.PostfetchExample(some_int=i)
-        session.add(postfetchExample)
+    examples = [models.PostfetchExample(some_int=i) for i in range(5)]
+    session.add_all(examples)
     session.flush()
-    assert postfetchExample.id == 4
+    assert sorted(pfe.id for pfe in examples) == [0, 1, 2, 3, 4]