def test_products(self): candidates = [DummyCandidateCluster([create_protocluster(0, 10)])] region = Region(candidate_clusters=candidates) assert region.products == ["a"] assert region.get_product_string() == "a" candidates = [] for i, prod in zip(range(2), "ba"): candidates.append(DummyCandidateCluster([create_protocluster(i*10, (i+1)*10, product=prod)])) region = Region(candidate_clusters=candidates) assert region.products == ["b", "a"] assert region.get_product_string() == "a,b"
def test_probabilities(self): loc = FeatureLocation(0, 10) candidates = [DummyCandidateCluster([create_protocluster(0, 10)])] assert Region(candidate_clusters=candidates).probabilities == [] subs = [SubRegion(loc, "testtool", probability=None)] assert Region(candidate_clusters=candidates, subregions=subs).probabilities == [] subs.append(SubRegion(loc, "testtool", probability=0.1)) assert Region(candidate_clusters=candidates, subregions=subs).probabilities == [0.1] subs.append(SubRegion(loc, "testtool", probability=0.7)) assert Region(candidate_clusters=candidates, subregions=subs).probabilities == [0.1, 0.7]
def setUp(self): self.protocluster = DummyProtocluster() self.candidate = DummyCandidateCluster([self.protocluster]) self.sub = SubRegion(self.protocluster.location, "testtool") self.region = Region(candidate_clusters=[self.candidate], subregions=[self.sub])