def test_union1(): gs1 = GeneSignature(name="test1", gene2weight=['TP53', 'SOX4']) gs2 = GeneSignature(name="test1", gene2weight=['TP53', 'SOX2']) gsu = gs1.union(gs2) assert 'TP53' in gsu assert 'SOX4' in gsu assert 'SOX2' in gsu assert len(gsu) == 3
def test_union3(): gs1 = GeneSignature(name="test1", gene2weight={'TP53': 0.8, 'SOX4': 0.75}) gs2 = GeneSignature(name="test1", gene2weight={'TP53': 0.3, 'SOX2': 0.60}) gsu = gs1.union(gs2) assert 'TP53' in gsu assert gsu.gene2weight['TP53'] == 0.8 assert 'SOX4' in gsu assert gsu.gene2weight['SOX4'] == 0.75 assert 'SOX2' in gsu assert gsu.gene2weight['SOX2'] == 0.6 assert len(gsu) == 3