def test_HMM_load(self): for mathType in self.mathTypes: hmm = HMM(mathType) hmm.load(self.inputHMMData[mathType]) X = hmm.toJSON() Y = self.inputHMMData[mathType] for i in range(len(Y["states"])): Y["states"][i] = Y["states"][i].toJSON() X["transitions"] = sorted(X["transitions"]) Y["transitions"] = sorted(Y["transitions"]) self.assertDictEqual(X, Y, "HMM.load/toJSON() is broken: " + \ str(X) + " != " + str(Y))