def testEqual(self):
diff = DiffFlameGraph()
compare = {"a;b;c": CsvRecord(50, "id1", "a;b;c"), "a;b": CsvRecord(int(-50), "id2", "a;b")}
compare2 = {"a;b;c": CsvRecord(50, "id1", "a;b;c"), "a;b": CsvRecord(int(-50), "id2", "a;b")}
compare3 = {"a;b;c": CsvRecord(10, "id1", "a;b;c"), "a;b": CsvRecord(int(-50), "id2", "a;b")}
compare4 = {"a;": CsvRecord(50, "id3", "a;"), "a;b": CsvRecord(int(-50), "id2", "a;b")}
self.assertTrue(diff.equal(compare, compare2))
self.assertFalse(diff.equal(compare, compare3))
self.assertFalse(diff.equal(compare, compare4))
def testDiffKeys(self):
diff = DiffFlameGraph()
val1 = diff.loadCSV("testa.txt")
val2 = diff.loadCSV("testb.txt")
diffCsv = diff.diffCsv(val1, val2)
# a;b; is gone from version 2 so should be -50
# a;b;c is new, so should be 50
compare = {"a;b;c": CsvRecord(50, "id1", "a;b;c"), "a;b": CsvRecord(int(-50), "id2", "a;b")}
self.assertTrue(diff.equal(diffCsv, compare))
def testWriteDiff(self):
diff = DiffFlameGraph()
val1 = diff.loadCSV("testa.txt")
val2 = diff.loadCSV("testb.txt")
diffCsv = diff.diffCsv(val1, val2)
diff.writeDiff(diffCsv, "../output/diff.txt")
diffCsv2 = diff.loadCSV("../output/diff.txt")
self.assertTrue(diff.equal(diffCsv, diffCsv2))
