def testOutputGraphOriginal():
def orderTuples(itms):
return list(
map(lambda x: (x[0], x[1]) if x[0] < x[1] else (x[1], x[0]), itms))
mca = MedeinaCumulativeApplication("dir")
mca.interactionStore = [
("Spec1", "Spec2"),
("Spec2", "Spec3"),
("Spec3", "Spec4"),
("Spec1", "Spec2"),
("Spec5", "Spec6"),
]
mca.scientificToUser = {
"Spec1": "USpec1",
"Spec2": "USpec2",
"Spec3": "USpec3",
"Spec4": "USpec4",
"Spec5": "USpec5",
"Spec6": "USpec6",
}
expected = [
("USpec1", "USpec2"),
("USpec2", "USpec3"),
("USpec3", "USpec4"),
("USpec5", "USpec6"),
]
unittest.TestCase().assertCountEqual(
orderTuples(mca.to_graph_original().edges), orderTuples(expected))
def testOutputListWithUserMapping():
mca = MedeinaCumulativeApplication("dir")
mca.interactionStore = [
("Spec1", "Spec2"),
("Spec2", "Spec3"),
("Spec3", "Spec4"),
("Spec1", "Spec2"),
("Spec5", "Spec6"),
]
mca.scientificToUser = {
"Spec1": "USpec1",
"Spec2": "USpec2",
"Spec3": "USpec3",
"Spec4": "USpec4",
"Spec5": "USpec5",
"Spec6": "USpec6",
}
expectedOne = [
("Spec1", "Spec2"),
("Spec2", "Spec3"),
("Spec3", "Spec4"),
("Spec5", "Spec6"),
]
expectedTwo = [
("USpec1", "USpec2"),
("USpec2", "USpec3"),
("USpec3", "USpec4"),
("USpec5", "USpec6"),
]
expected = list(zip(expectedOne, expectedTwo))
unittest.TestCase().assertCountEqual(mca.to_list_original(), expected)
def testOutputMatrixOriginal():
def orderTuples(itms):
return list(
map(lambda x: (x[0], x[1]) if x[0] < x[1] else (x[1], x[0]), itms))
mca = MedeinaCumulativeApplication("dir")
mca.interactionStore = [
("Spec1", "Spec2"),
("Spec2", "Spec3"),
("Spec3", "Spec4"),
("Spec1", "Spec2"),
("Spec5", "Spec6"),
]
mca.scientificToUser = {
"Spec1": "USpec1",
"Spec2": "USpec2",
"Spec3": "USpec3",
"Spec4": "USpec4",
"Spec5": "USpec5",
"Spec6": "USpec6",
}
crushed = []
matrixOut, labelOrder = mca.to_matrix_original()
matrixOut = matrixOut.tolist()
for kr, row in enumerate(matrixOut):
for kc, col in enumerate(row):
if col == 1:
crushed.append((labelOrder[kr], labelOrder[kc]))
crushed = list(set(orderTuples(crushed)))
expected = [
("USpec1", "USpec2"),
("USpec2", "USpec3"),
("USpec3", "USpec4"),
("USpec5", "USpec6"),
]
unittest.TestCase().assertCountEqual(orderTuples(expected), crushed)