implement pmx.writer

[meshio/pymeshio.git] / test / pmx_test.py

diff --git a/test/pmx_test.py b/test/pmx_test.py
index 1e89b0d..e9185ee 100644 (file)
--- a/test/pmx_test.py
+++ b/test/pmx_test.py
@@ -1,7 +1,9 @@
 # coding: utf-8\r
 import unittest\r
+import io\r
 import pymeshio.pmd\r
 import pymeshio.pmx.reader\r
+import pymeshio.pmx.writer\r
 \r
 \r
 PMX_FILE=pymeshio.unicode('resources/初音ミクVer2.pmx')\r
@@ -44,3 +46,14 @@ class TestPmx(unittest.TestCase):
         self.assertEqual(45,  len(model.rigidbodies))\r
         self.assertEqual(27,  len(model.joints))\r
 \r
+    def test_write(self):\r
+        # read source file\r
+        buf=pymeshio.common.readall(PMX_FILE)\r
+        # read and write to out\r
+        model=pymeshio.pmx.reader.read(io.BytesIO(buf))\r
+        out=io.BytesIO()\r
+        pymeshio.pmx.writer.write(out, model)\r
+        # read out buffer again\r
+        model2=pymeshio.pmx.reader.read(io.BytesIO(out.getvalue()))\r
+        self.assertEqual(model, model2)\r
+\r