]> granicus.if.org Git - python/commitdiff
Add test of file.write(array) extracted from Jython.
authorFrank Wierzbicki <fwierzbicki@gmail.com>
Sun, 16 Aug 2009 20:22:51 +0000 (20:22 +0000)
committerFrank Wierzbicki <fwierzbicki@gmail.com>
Sun, 16 Aug 2009 20:22:51 +0000 (20:22 +0000)
Lib/test/test_array.py

index 0ef5f9e518bc23ae94fbec6cd80dc5eab0a9007a..63f7083d337341c5827771f94021b5b40a4e2565 100755 (executable)
@@ -188,6 +188,25 @@ class BaseTest(unittest.TestCase):
                 f.close()
             test_support.unlink(test_support.TESTFN)
 
+    def test_filewrite(self):
+        a = array.array(self.typecode, 2*self.example)
+        f = open(test_support.TESTFN, 'wb')
+        try:
+            f.write(a)
+            f.close()
+            b = array.array(self.typecode)
+            f = open(test_support.TESTFN, 'rb')
+            b.fromfile(f, len(self.example))
+            self.assertEqual(b, array.array(self.typecode, self.example))
+            self.assertNotEqual(a, b)
+            b.fromfile(f, len(self.example))
+            self.assertEqual(a, b)
+            f.close()
+        finally:
+            if not f.closed:
+                f.close()
+            test_support.unlink(test_support.TESTFN)
+
     def test_tofromlist(self):
         a = array.array(self.typecode, 2*self.example)
         b = array.array(self.typecode)