def test_remove_parent_child_and_parent_and_sibling_in_remain(self):
# given
remove = pandas.DataFrame({
'id1': ['a'],
'edgeType': [EdgeType.GENE_REACTION_GENE],
'id2': ["0"],
'qscore': "100",
'value': list('x')
})
remain = pandas.DataFrame({
'id1': ['a', 'a', 'a'],
'edgeType': [
EdgeType.GENE_GENE, EdgeType.GENE_EXPRESSION_GENE,
EdgeType.GENE_REACTION_GENE
],
'id2': ["0", "0", "0"],
'qscore': [100.0, 50, 20],
'value':
list('xxx')
})
# when
result = utils.remove_parent_duplicates_and_reverses(remove_set=remove,
remain_set=remain)
# then
true_result = pandas.DataFrame({
'id1': ['a', 'a'],
'edgeType':
[EdgeType.GENE_EXPRESSION_GENE, EdgeType.GENE_REACTION_GENE],
'id2': ["0", "0"],
'qscore': [50, 20],
'value':
list('xx')
})
np.testing.assert_array_equal(true_result.values, result.values)
def test_remove_parent_child_and_parent_and_sibling_in_remain(self):
# given
remove = pandas.DataFrame({
"id1": ["a"],
"edgeType": [EdgeType.GENE_REACTION_GENE],
"id2": ["0"],
"qscore": "100",
"value": list("x")
})
remain = pandas.DataFrame({
"id1": ["a", "a", "a"],
"edgeType": [
EdgeType.GENE_GENE, EdgeType.GENE_EXPRESSION_GENE,
EdgeType.GENE_REACTION_GENE
],
"id2": ["0", "0", "0"],
"qscore": [100.0, 50, 20],
"value":
list("xxx"),
})
# when
result = utils.remove_parent_duplicates_and_reverses(remove_set=remove,
remain_set=remain)
# then
true_result = pandas.DataFrame({
"id1": ["a", "a"],
"edgeType":
[EdgeType.GENE_EXPRESSION_GENE, EdgeType.GENE_REACTION_GENE],
"id2": ["0", "0"],
"qscore": [50, 20],
"value":
list("xx"),
})
np.testing.assert_array_equal(true_result.values, result.values)
def test_remove_parent_reverse_parent(self):
#given
remove = pandas.DataFrame({
'id1': ['a'],
'edgeType': [EdgeType.GENE_REACTION_GENE],
'id2': ["0"],
'qscore': ["a"],
'value': list('x')
})
remain = pandas.DataFrame({
'id1': ['0', 'b', 'c'],
'edgeType':
[EdgeType.GENE_GENE, EdgeType.GENE_GENE, EdgeType.GENE_GENE],
'id2': ["a", "1", "2"],
'qscore': [100.0, 50, 20],
'value':
list('xxy')
})
#when
result = utils.remove_parent_duplicates_and_reverses(remove_set=remove,
remain_set=remain)
#then
true_result = pandas.DataFrame({
'id1': ['b', 'c'],
'edgeType': [EdgeType.GENE_GENE, EdgeType.GENE_GENE],
'id2': ["1", "2"],
'qscore': [50, 20],
'value':
list('xy')
})
np.testing.assert_array_equal(true_result.values, result.values)
def test_remove_parent_parent_in_remain(self):
# given
remove = pandas.DataFrame({
"id1": ["a"],
"edgeType": [EdgeType.GENE_REACTION_GENE],
"id2": ["0"],
"qscore": "100",
"value": [1]
})
remain = pandas.DataFrame({
"id1": ["a", "b", "c"],
"edgeType":
[EdgeType.GENE_GENE, EdgeType.GENE_GENE, EdgeType.GENE_GENE],
"id2": ["0", "1", "2"],
"qscore": [120.0, 50, 20],
"value": [1, 1, 0],
})
# when
result = utils.remove_parent_duplicates_and_reverses(remove_set=remove,
remain_set=remain)
# then
true_result = pandas.DataFrame({
"id1": ["b", "c"],
"edgeType": [EdgeType.GENE_GENE, EdgeType.GENE_GENE],
"id2": ["1", "2"],
"qscore": [50, 20],
"value": [1, 0],
})
np.testing.assert_array_equal(true_result.values, result.values)