Handle duplicate Index values in data frame
Adding a test case for checking if duplicates are
handled properly
Change-Id: Ib04be8b414e16e164e93b8beaa33bffd4b40abe7
Signed-off-by: Kapileshwar Singh <kapileshwar.singh@arm.com>
diff --git a/tests/test_duplicates.py b/tests/test_duplicates.py
new file mode 100644
index 0000000..fbdc8da
--- /dev/null
+++ b/tests/test_duplicates.py
@@ -0,0 +1,84 @@
+#!/usr/bin/env python
+
+import unittest
+import matplotlib
+import pandas as pd
+import utils_tests
+import cr2
+import shutil
+
+from test_thermal import BaseTestThermal
+
+
+class TestPlotterDupVals(BaseTestThermal):
+
+ """Test Duplicate Entries in plotter"""
+
+ def __init__(self, *args, **kwargs):
+ super(TestPlotterDupVals, self).__init__(*args, **kwargs)
+
+ def test_plotter_duplicates(self):
+ """Test that plotter handles duplicates fine"""
+ with open("trace.txt", "w") as fout:
+ fout.write("""version = 6
+cpus=6
+rcuos/2-22 [001] 0000.018510: sched_load_avg_sg: cpus=00000001 load=0 utilization=0
+rcuos/2-22 [001] 6550.018611: sched_load_avg_sg: cpus=00000002 load=1 utilization=1
+rcuos/2-22 [001] 6550.018611: sched_load_avg_sg: cpus=00000004 load=2 utilization=2
+rcuos/2-22 [001] 6550.018612: sched_load_avg_sg: cpus=00000001 load=2 utilization=3
+rcuos/2-22 [001] 6550.018624: sched_load_avg_sg: cpus=00000002 load=1 utilization=4
+rcuos/2-22 [001] 6550.018625: sched_load_avg_sg: cpus=00000002 load=2 utilization=5
+rcuos/2-22 [001] 6550.018626: sched_load_avg_sg: cpus=00000002 load=3 utilization=6
+rcuos/2-22 [001] 6550.018627: sched_load_avg_sg: cpus=00000002 load=1 utilization=7
+rcuos/2-22 [001] 6550.018628: sched_load_avg_sg: cpus=00000004 load=2 utilization=8\n""")
+ fout.close()
+ run1 = cr2.Run(name="first")
+ l = cr2.LinePlot(
+ run1,
+ cr2.sched.SchedLoadAvgSchedGroup,
+ column=['utilization'],
+ filters={
+ "load": [
+ 1,
+ 2]},
+ pivot="cpus",
+ marker='o',
+ linestyle='none',
+ per_line=3)
+ l.view()
+ matplotlib.pyplot.close('all')
+
+ def test_plotter_triplicates(self):
+
+ """Test that plotter handles triplicates fine"""
+
+ with open("trace.txt", "w") as fout:
+ fout.write("""version = 6
+cpus=6
+rcuos/2-22 [001] 0000.018510: sched_load_avg_sg: cpus=00000001 load=0 utilization=0
+rcuos/2-22 [001] 6550.018611: sched_load_avg_sg: cpus=00000002 load=1 utilization=1
+rcuos/2-22 [001] 6550.018611: sched_load_avg_sg: cpus=00000004 load=2 utilization=2
+rcuos/2-22 [001] 6550.018611: sched_load_avg_sg: cpus=00000004 load=2 utilization=2
+rcuos/2-22 [001] 6550.018612: sched_load_avg_sg: cpus=00000001 load=2 utilization=3
+rcuos/2-22 [001] 6550.018624: sched_load_avg_sg: cpus=00000002 load=1 utilization=4
+rcuos/2-22 [001] 6550.018625: sched_load_avg_sg: cpus=00000002 load=2 utilization=5
+rcuos/2-22 [001] 6550.018626: sched_load_avg_sg: cpus=00000002 load=3 utilization=6
+rcuos/2-22 [001] 6550.018627: sched_load_avg_sg: cpus=00000002 load=1 utilization=7
+rcuos/2-22 [001] 6550.018628: sched_load_avg_sg: cpus=00000004 load=2 utilization=8\n""")
+ fout.close()
+
+ run1 = cr2.Run(name="first")
+ l = cr2.LinePlot(
+ run1,
+ cr2.sched.SchedLoadAvgSchedGroup,
+ column=['utilization'],
+ filters={
+ "load": [
+ 1,
+ 2]},
+ pivot="cpus",
+ marker='o',
+ linestyle='none',
+ per_line=3)
+ l.view()
+ matplotlib.pyplot.close('all')