summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorJohn Ehresman <jpe@wingware.com>2010-04-16 11:55:50 -0400
committerDavid Malcolm <dmalcolm@redhat.com>2010-04-16 13:02:42 -0400
commitcecafa7e4809c8c2bfa77446beea8beb3a51d8dc (patch)
tree00d4bb95a55f3e55dc4c0d453f75011852667405
parentbe40a7e479e12e6c5f15603c134c926ac300ba7b (diff)
downloadpygobject-cecafa7e4809c8c2bfa77446beea8beb3a51d8dc.tar.gz
pygobject-cecafa7e4809c8c2bfa77446beea8beb3a51d8dc.tar.xz
pygobject-cecafa7e4809c8c2bfa77446beea8beb3a51d8dc.zip
Test fixes for bytes
-rw-r--r--tests/common.py5
-rw-r--r--tests/test_gicon.py12
-rw-r--r--tests/test_mainloop.py4
3 files changed, 12 insertions, 9 deletions
diff --git a/tests/common.py b/tests/common.py
index 5222230..b5f5bfc 100644
--- a/tests/common.py
+++ b/tests/common.py
@@ -51,7 +51,10 @@ def importModule(module, directory, name=None):
if sys.version_info >= (3, 0):
def _bytes(s):
- return s.encode('latin1')
+ if isinstance(s, str):
+ return s.encode('latin1')
+ else:
+ return bytes(s)
else:
_bytes = str
diff --git a/tests/test_gicon.py b/tests/test_gicon.py
index 5ce7ece..67e33eb 100644
--- a/tests/test_gicon.py
+++ b/tests/test_gicon.py
@@ -3,7 +3,7 @@
import os
import unittest
-from common import gio, glib
+from common import gio, glib, _bytes
class TestIcon(unittest.TestCase):
@@ -28,11 +28,11 @@ class TestIcon(unittest.TestCase):
class TestLoadableIcon(unittest.TestCase):
def setUp(self):
- self.file = open('temp.svg', 'w')
- self.svg = ('<?xml version="1.0" encoding="UTF-8" standalone="no"?>'
- '<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.0//EN" '
- '"http://www.w3.org/TR/2001/REC-SVG-20010904/DTD/svg10.dtd">'
- '<svg width="32" height="32"/>')
+ self.file = open('temp.svg', 'wb')
+ self.svg = _bytes('<?xml version="1.0" encoding="UTF-8" standalone="no"?>'
+ '<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.0//EN" '
+ '"http://www.w3.org/TR/2001/REC-SVG-20010904/DTD/svg10.dtd">'
+ '<svg width="32" height="32"/>')
self.file.write(self.svg)
self.file.close()
self.icon = gio.FileIcon(gio.File('temp.svg'))
diff --git a/tests/test_mainloop.py b/tests/test_mainloop.py
index afb550a..a82063f 100644
--- a/tests/test_mainloop.py
+++ b/tests/test_mainloop.py
@@ -5,7 +5,7 @@ import sys
import select
import unittest
-from common import glib
+from common import glib, _bytes
class TestMainLoop(unittest.TestCase):
def testExceptionHandling(self):
@@ -26,7 +26,7 @@ class TestMainLoop(unittest.TestCase):
glib.child_watch_add(pid, child_died, loop)
os.close(pipe_r)
- os.write(pipe_w, "Y")
+ os.write(pipe_w, _bytes("Y"))
os.close(pipe_w)
def excepthook(type, value, traceback):