def setUp(self):
self.mu1 = np.array([0, 0, 0])
self.sig = 2 * np.eye(3)
self.mu2 = np.array([5, 5, 5])
self.clust1 = DPCluster(.5, self.mu1, self.sig)
self.clust2 = DPCluster(.5, self.mu2, self.sig)
self.clusters = [self.clust1, self.clust2]
self.x = DPMixture(self.clusters, niter=1, identified=True)
self.diag = DiagonalAlignData(self.x, size=100000)
self.comp = CompAlignData(self.x, size=100000)
self.full = FullAlignData(self.x, size=200000)
