Re-added missing test.

dessibelle · Feb 17, 2015 · 3e4ad17 · 3e4ad17
1 parent 4d25022
commit 3e4ad17
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/payer_api/tests/test_xml.py b/payer_api/tests/test_xml.py
@@ -195,6 +195,7 @@ def test_urls(self):
 
     def test_xml(self):
         xml = self.xml_document.tostring()
+        self.assertEqual(str(self.xml_document), xml)
 
         try:
             parser = etree.XMLParser(dtd_validation=False)

diff --git a/payer_api/xml.py b/payer_api/xml.py
@@ -149,7 +149,7 @@ def tostring(self, encoding="utf-8", pretty_print=False,
         tree.write(output, pretty_print=pretty_print, xml_declaration=True,
                    encoding=encoding, method="xml")
 
-        retval = output.getvalue().decode('utf-8')
+        retval = output.getvalue().decode(encoding)
         output.close()
 
         return retval