def test_formatResultCsv():
expected = {
"perdictions": [
"merge",
"dup",
"none",
],
"ids": [
{
"lhs_id": '210204',
"rhs_id": '210202'
},
{
"lhs_id": '210205',
"rhs_id": '210202'
},
{
"lhs_id": '210204',
"rhs_id": '210208'
},
]
}
inputCsv = [
[
"Dup?", "Merge?", "Reason", "Merged in Moztrap?", "Case ID 1",
"Case ID 2", "Case 1 ", "Case 2", "Diff", "CaseVersion ID 1",
"CaseVersion ID 2"
],
[
"No", "Yes", "", "", "", "", "", "",
"http://shinglyu.github.io/moztrap-new-ui/diff.html?lhs=210204&rhs=210202",
"210204", "210202"
],
[
"Yes", "No", "", "", "", "", "", "",
"http://shinglyu.github.io/moztrap-new-ui/diff.html?lhs=210205&rhs=210202",
"210205", "210202"
],
[
"No", "No", "", "", "", "", "", "",
"http://shinglyu.github.io/moztrap-new-ui/diff.html?lhs=210204&rhs=210208",
"210204", "210208"
],
]
#expected = [
#"Dup?,Merge?,Reason,Merged in Moztrap?,Case ID 1,Case ID 2,Case 1 ,Case 2,Diff,CaseVersion ID 1,CaseVersion ID 2",
#"No,Yes,,,706,707,https://moztrap.mozilla.org/manage/cases/?filter-id=706,https://moztrap.mozilla.org/manage/cases/?filter-id=707,http://shinglyu.github.io/moztrap-new-ui/diff.html?lhs=210201&rhs=210202,210204,210202",
#"Yes,No,,,706,1696,https://moztrap.mozilla.org/manage/cases/?filter-id=706,https://moztrap.mozilla.org/manage/cases/?filter-id=1696,http://shinglyu.github.io/moztrap-new-ui/diff.html?lhs=210201&rhs=210521,210205,210202"
#"No,No,,,706,1696,https://moztrap.mozilla.org/manage/cases/?filter-id=706,https://moztrap.mozilla.org/manage/cases/?filter-id=1696,http://shinglyu.github.io/moztrap-new-ui/diff.html?lhs=210201&rhs=210521,210204,210208"
#]
assert (expected == output.parseResultCsv(inputCsv))
def loadGroundTruth(filename, caseversions=[]):
existing_case_ids = map(lambda x:str(x['id']), caseversions)
# TODO: move this parsing part to output.py
with open(filename, 'r') as csvfile:
rows = csv.reader(csvfile, delimiter=",", quotechar="\"")
gt = output.parseResultCsv(rows)
ids = []
targets = [] # answers
if len(existing_case_ids) > 0:
for idx in range(len(gt['ids'])):
case1 = gt['ids'][idx]['lhs_id']
case2 = gt['ids'][idx]['rhs_id']
if (case1 in existing_case_ids and case2 in existing_case_ids):
targets.append(gt['perdictions'][idx])
ids.append(gt['ids'][idx])
else:
ids = gt['ids']
targets = gt['perdictions']
return {'ids': ids, 'perdictions': targets}
def loadGroundTruth(filename, caseversions=[]):
existing_case_ids = map(lambda x: str(x['id']), caseversions)
# TODO: move this parsing part to output.py
with open(filename, 'r') as csvfile:
rows = csv.reader(csvfile, delimiter=",", quotechar="\"")
gt = output.parseResultCsv(rows)
ids = []
targets = [] # answers
if len(existing_case_ids) > 0:
for idx in range(len(gt['ids'])):
case1 = gt['ids'][idx]['lhs_id']
case2 = gt['ids'][idx]['rhs_id']
if (case1 in existing_case_ids and case2 in existing_case_ids):
targets.append(gt['perdictions'][idx])
ids.append(gt['ids'][idx])
else:
ids = gt['ids']
targets = gt['perdictions']
return {'ids': ids, 'perdictions': targets}
def test_formatResultCsv():
expected = {
"perdictions": [
"merge",
"dup",
"none",
],
"ids": [
{
"lhs_id": '210204',
"rhs_id": '210202'
},
{
"lhs_id": '210205',
"rhs_id": '210202'
},
{
"lhs_id": '210204',
"rhs_id": '210208'
},
]
}
inputCsv = [
["Dup?","Merge?","Reason","Merged in Moztrap?","Case ID 1","Case ID 2","Case 1 ","Case 2","Diff","CaseVersion ID 1","CaseVersion ID 2"],
["No","Yes","","","","","","","http://shinglyu.github.io/moztrap-new-ui/diff.html?lhs=210204&rhs=210202","210204","210202"],
["Yes","No","","","","","","","http://shinglyu.github.io/moztrap-new-ui/diff.html?lhs=210205&rhs=210202","210205","210202"],
["No","No","","","","","","","http://shinglyu.github.io/moztrap-new-ui/diff.html?lhs=210204&rhs=210208","210204","210208"],
]
#expected = [
#"Dup?,Merge?,Reason,Merged in Moztrap?,Case ID 1,Case ID 2,Case 1 ,Case 2,Diff,CaseVersion ID 1,CaseVersion ID 2",
#"No,Yes,,,706,707,https://moztrap.mozilla.org/manage/cases/?filter-id=706,https://moztrap.mozilla.org/manage/cases/?filter-id=707,http://shinglyu.github.io/moztrap-new-ui/diff.html?lhs=210201&rhs=210202,210204,210202",
#"Yes,No,,,706,1696,https://moztrap.mozilla.org/manage/cases/?filter-id=706,https://moztrap.mozilla.org/manage/cases/?filter-id=1696,http://shinglyu.github.io/moztrap-new-ui/diff.html?lhs=210201&rhs=210521,210205,210202"
#"No,No,,,706,1696,https://moztrap.mozilla.org/manage/cases/?filter-id=706,https://moztrap.mozilla.org/manage/cases/?filter-id=1696,http://shinglyu.github.io/moztrap-new-ui/diff.html?lhs=210201&rhs=210521,210204,210208"
#]
assert(expected == output.parseResultCsv(inputCsv))
