test/pmd_test.py

   1 # coding: utf-8
   2 import sys
   3 import io
   4 import unittest
   5 import pymeshio.pmd
   6 import pymeshio.pmd.reader
   7 import pymeshio.pmd.writer
   8
   9
  10 PMD_FILE=pymeshio.unicode('resources/初音ミクVer2.pmd')
  11
  12
  13 class TestPmd(unittest.TestCase):
  14
  15     def setUp(self):
  16         pass
  17
  18     def test_read(self):
  19         model=pymeshio.pmd.reader.read_from_file(PMD_FILE)
  20         self.assertEqual(pymeshio.pmd.Model,  model.__class__)
  21         self.assertEqual(pymeshio.unicode('初音ミク').encode('cp932'),  model.name)
  22         self.assertEqual(pymeshio.unicode('Miku Hatsune').encode('cp932'),  model.english_name)
  23         self.assertEqual(pymeshio.unicode(
  24             "PolyMo用モデルデータ：初音ミク ver.2.3\n"+
  25             "(物理演算対応モデル)\n"+
  26             "\n"+
  27             "モデリング    ：あにまさ氏\n"+
  28             "データ変換    ：あにまさ氏\n"+
  29             "Copyright  ：CRYPTON FUTURE MEDIA, INC").encode('cp932'),
  30             model.comment)
  31         self.assertEqual(pymeshio.unicode(
  32             "MMD Model: Miku Hatsune ver.2.3\n"+
  33             "(Physical Model)\n"+
  34             "\n"+
  35             "Modeling by        Animasa\n"+
  36             "Converted by       Animasa\n"+
  37             "Copyright          CRYPTON FUTURE MEDIA, INC").encode('cp932'),
  38             model.english_comment)
  39         self.assertEqual(12354,  len(model.vertices))
  40         self.assertEqual(22961 * 3,  len(model.indices))
  41         print("{0} textures".format(len(model.toon_textures)))
  42         self.assertEqual(17,  len(model.materials))
  43         self.assertEqual(140,  len(model.bones))
  44         self.assertEqual(31,  len(model.morphs))
  45         self.assertEqual(45,  len(model.rigidbodies))
  46         self.assertEqual(27,  len(model.joints))
  47
  48     def test_write(self):
  49         # read source file
  50         buf=pymeshio.common.readall(PMD_FILE)
  51         # read and write to out
  52         model=pymeshio.pmd.reader.read(io.BytesIO(buf))
  53         out=io.BytesIO()
  54         pymeshio.pmd.writer.write(out, model)
  55         # read out buffer again
  56         model2=pymeshio.pmd.reader.read(io.BytesIO(out.getvalue()))
  57         model.diff(model2)
  58         self.assertEqual(model, model2)
  59
  60