These get easier to test with Python 3

circle.yml

@@ -4,4 +4,4 @@ machine:
 test:
   override:
     - bundle exec rake
-    - python setup.py test
+    - python3 setup.py test

metroextractor/nothingburger.py

@@ -1,5 +1,6 @@
 from __future__ import print_function
 from argparse import ArgumentParser
+import io
 
 parser = ArgumentParser(description='Tell us what they told you.')
 
@@ -12,7 +13,7 @@
 def tell_us(bbox, filename):
     '''
     '''
-    with open(filename, 'w') as file:
+    with io.open(filename, 'w') as file:
         print(bbox, file=file)
 
 def main():

metroextractor/tests.py

@@ -1,17 +1,13 @@
-import unittest, tempfile, os
+import unittest, unittest.mock
 
 from . import nothingburger
 
 class Nothingburger (unittest.TestCase):
 
-    def setUp(self):
-        handle, self.filename = tempfile.mkstemp()
-        os.close(handle)
-
-    def tearDown(self):
-        os.remove(self.filename)
-
     def test_tell_us(self):
-        nothingburger.tell_us([0, 1, 2, 3], self.filename)
-        with open(self.filename) as file:
-            self.assertEqual(next(file), '[0, 1, 2, 3]\n')
+        with unittest.mock.patch('io.open') as io_open:
+            nothingburger.tell_us([0, 1, 2, 3], 'filename.txt')
+
+        io_open.assert_called_once_with('filename.txt', 'w')
+        write_call = io_open.return_value.__enter__.return_value.mock_calls[0]
+        self.assertTrue(write_call[1][0].startswith('[0, 1, 2, 3]'))

recipes/scripts.rb

@@ -1,4 +1,4 @@
-execute 'pip install -U .' do
+execute 'pip3 install -U .' do
   cwd File.join(File.dirname(__FILE__), '..')
   environment('LC_ALL' => 'C.UTF-8') # ...for correct encoding in python open()
 end