Global variables are evil. Tests under both py2k and py3k pass.

2 files changed, 2 insertions, 5 deletions

diff --git a/test/test_pyg.py b/test/test_pyg.py
index 4076fc3..ddf8146 100755
--- a/test/test_pyg.py
+++ b/test/test_pyg.py
@@ -1,6 +1,4 @@
 #!/usr/bin/python
-# -*- coding: utf-8 -*-
-
 import re
 import subprocess
 import unittest
diff --git a/test/test_wlp.py b/test/test_wlp.py
index 430c235..929fb1c 100755
--- a/test/test_wlp.py
+++ b/test/test_wlp.py
@@ -1,5 +1,4 @@
 #!/usr/bin/python
-# -*- coding: utf-8 -*-
 import unittest
 import wlp_parser
 
@@ -97,8 +96,8 @@ class TestWLP(unittest.TestCase):
         self.assertEqual(tree, expected_tree)
 
     def test_wlp(self):
-        wlp.setfilebyname('examples/whitelist.example')
-        wl_dict = wlp.mkdict()
+        with open('examples/whitelist.example') as inf:
+            wl_dict = wlp.mkdict(inf)
         self.assertEqual(wl_dict, EXPECTED_WL)