test madness

twneale · Sep 6, 2013 · f110170 · f110170
1 parent 4156124
commit f110170
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/tests/test_rollcall.py b/tests/test_rollcall.py
@@ -29,10 +29,9 @@ class RollcallTest(TestCase):
     # origininall created.
     here = dirname(abspath(__file__))
     with cd(join(here, 'fixtures')):
-        f = open('sen109kh.ord')
-    ordfile = OrdFile(f)
+        with open('sen109kh.ord') as f:
+            rollcall = Rollcall.from_ordfile(f)
 
-    rollcall = ordfile.as_rollcall()
     summary = rollcall.summary()
 
     def test_summary(self):

diff --git a/tests/test_wnominate.py b/tests/test_wnominate.py
@@ -32,10 +32,9 @@ class TestNomNom(TestCase):
     # origininall created.
     here = dirname(abspath(__file__))
     with cd(join(here, 'fixtures')):
-        f = open('sen90kh.ord')
-    ordfile = OrdFile(f)
+        with open('sen90kh.ord') as f:
+            rollcall = Rollcall.from_ordfile(f)
 
-    rollcall = ordfile.as_rollcall()
     summary = rollcall.summary()
 
     # Calculate the expected wnominate result.