def test_proto(self):
h1 = HtkIO()
h1.write_hmm_proto( 25, os.path.join(HERE,"proto_from_htkio") )
h2 = HMM()
h2.create_proto( 25 )
h2.save( os.path.join(HERE,"proto_from_hmm") )
m1 = HMM()
m1.load( os.path.join(HERE,"proto_from_htkio") )
m2 = HMM()
m2.load( os.path.join(HERE,"proto_from_hmm") )
self.assertTrue(compare(m1.definition['transition'],m2.definition['transition']))
self.assertTrue(compare(m1.definition['states'],m2.definition['states']))
os.remove( os.path.join(HERE,"proto_from_hmm") )
os.remove( os.path.join(HERE,"proto_from_htkio") )
def testMix(self):
acmodel1 = AcModel()
hmm1 = HMM()
hmm1.create_proto( 25 )
hmm1.name = "y"
acmodel1.append_hmm( hmm1 )
acmodel1.repllist.add("y","j")
acmodel2 = AcModel()
hmm2 = HMM()
hmm2.create_proto( 25 )
hmm2.name = "j"
hmm3 = HMM()
hmm3.create_proto( 25 )
hmm3.name = "y"
acmodel2.hmms.append( hmm2 )
acmodel2.hmms.append( hmm3 )
acmodel2.repllist.add("y","y")
acmodel2.repllist.add("j","j")
modelmixer = ModelMixer()
modelmixer.set_models( acmodel1,acmodel2 )
outputdir = os.path.join(MODELDIR, "models-test")
modelmixer.mix( outputdir, gamma=1. )
mixedh1 = AcModel()
mixedh1.load( outputdir )
shutil.rmtree( outputdir )
