Python3: minor fixes

twoolie · Mar 13, 2012 · 0ba7890 · 0ba7890
1 parent 8d050c0
commit 0ba7890
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/nbt/nbt.py b/nbt/nbt.py
@@ -8,8 +8,10 @@
 
 try:
 	unicode
+	basestring
 except NameError:
 	unicode = str  # compatibility for Python 3
+	basestring = str  # compatibility for Python 3
 
 
 TAG_END = 0

diff --git a/tests/tests.py b/tests/tests.py
@@ -22,7 +22,7 @@ def testEmptyFiles(self):
 		Opening an empty file causes an uncaught exception.
 		https://github.com/twoolie/NBT/issues/issue/4
 		"""
-		temp = BytesIO("")
+		temp = BytesIO(b"")
 		temp.seek(0)
 		self.assertRaises(MalformedFileError, NBTFile, buffer=temp)
 
@@ -102,7 +102,7 @@ def tearDown(self):
 class EmptyStringTest(unittest.TestCase):
 
 	def setUp(self):
-		self.golden_value = "\x0A\0\x04Test\x08\0\x0Cempty string\0\0\0"
+		self.golden_value = b"\x0A\0\x04Test\x08\0\x0Cempty string\0\0\0"
 		self.nbtfile = NBTFile(buffer=BytesIO(self.golden_value))
 
 	def testReadEmptyString(self):