def setUp(self): self.pis = array([[.1,.1,.8], [.1,.8,.1],[.8,.1,.1]]) self.mus = array([[2,2],[3,3],[4,4]]) self.sigmas = array([eye(2),eye(2),eye(2)]) self.mix = HDPMixture(self.pis,self.mus,self.sigmas)