| # Copyright 2012 Google Inc. All Rights Reserved. |
| """Tests for the tabulator module.""" |
| |
| from __future__ import print_function |
| |
| __author__ = 'asharif@google.com (Ahmad Sharif)' |
| |
| # System modules |
| import unittest |
| |
| # Local modules |
| import tabulator |
| |
| |
| class TabulatorTest(unittest.TestCase): |
| |
| def testResult(self): |
| table = ['k1', ['1', '3'], ['55']] |
| result = tabulator.Result() |
| cell = tabulator.Cell() |
| result.Compute(cell, table[2], table[1]) |
| expected = ' '.join([str(float(v)) for v in table[2]]) |
| self.assertTrue(cell.value == expected) |
| |
| result = tabulator.AmeanResult() |
| cell = tabulator.Cell() |
| result.Compute(cell, table[2], table[1]) |
| self.assertTrue(cell.value == float(table[2][0])) |
| |
| def testStringMean(self): |
| smr = tabulator.StringMeanResult() |
| cell = tabulator.Cell() |
| value = 'PASS' |
| values = [value for _ in range(3)] |
| smr.Compute(cell, values, None) |
| self.assertTrue(cell.value == value) |
| values.append('FAIL') |
| smr.Compute(cell, values, None) |
| self.assertTrue(cell.value == '?') |
| |
| def testStorageFormat(self): |
| sf = tabulator.StorageFormat() |
| cell = tabulator.Cell() |
| base = 1024.0 |
| cell.value = base |
| sf.Compute(cell) |
| self.assertTrue(cell.string_value == '1.0K') |
| cell.value = base**2 |
| sf.Compute(cell) |
| self.assertTrue(cell.string_value == '1.0M') |
| cell.value = base**3 |
| sf.Compute(cell) |
| self.assertTrue(cell.string_value == '1.0G') |
| |
| def testLerp(self): |
| c1 = tabulator.Color(0, 0, 0, 0) |
| c2 = tabulator.Color(255, 0, 0, 0) |
| c3 = tabulator.Color.Lerp(0.5, c1, c2) |
| self.assertTrue(c3.r == 127.5) |
| self.assertTrue(c3.g == 0) |
| self.assertTrue(c3.b == 0) |
| self.assertTrue(c3.a == 0) |
| c3.Round() |
| self.assertTrue(c3.r == 127) |
| |
| def testGmean(self): |
| a = [1.0e+308] * 3 |
| b = tabulator.Result()._GetGmean(a) |
| self.assertTrue(b >= 0.99e+308 and b <= 1.01e+308) |
| |
| def testTableGenerator(self): |
| runs = [[{'k1': '10', |
| 'k2': '12'}, {'k1': '13', |
| 'k2': '14', |
| 'k3': '15'}], [{'k1': '50', |
| 'k2': '51', |
| 'k3': '52', |
| 'k4': '53'}]] |
| labels = ['vanilla', 'modified'] |
| tg = tabulator.TableGenerator(runs, labels) |
| table = tg.GetTable() |
| header = table.pop(0) |
| |
| self.assertTrue(header == ['keys', 'vanilla', 'modified']) |
| row = table.pop(0) |
| self.assertTrue(row == ['k1', ['10', '13'], ['50']]) |
| row = table.pop(0) |
| self.assertTrue(row == ['k2', ['12', '14'], ['51']]) |
| row = table.pop(0) |
| self.assertTrue(row == ['k3', [None, '15'], ['52']]) |
| row = table.pop(0) |
| self.assertTrue(row == ['k4', [None, None], ['53']]) |
| |
| table = tg.GetTable() |
| columns = [ |
| tabulator.Column(tabulator.AmeanResult(), tabulator.Format()), |
| tabulator.Column(tabulator.AmeanRatioResult(), |
| tabulator.PercentFormat()), |
| ] |
| tf = tabulator.TableFormatter(table, columns) |
| table = tf.GetCellTable() |
| self.assertTrue(table) |
| |
| def testColspan(self): |
| simple_table = [ |
| ['binary', 'b1', 'b2', 'b3'], |
| ['size', 100, 105, 108], |
| ['rodata', 100, 80, 70], |
| ['data', 100, 100, 100], |
| ['debug', 100, 140, 60], |
| ] |
| columns = [ |
| tabulator.Column(tabulator.AmeanResult(), tabulator.Format()), |
| tabulator.Column(tabulator.MinResult(), tabulator.Format()), |
| tabulator.Column(tabulator.AmeanRatioResult(), |
| tabulator.PercentFormat()), |
| tabulator.Column(tabulator.AmeanRatioResult(), |
| tabulator.ColorBoxFormat()), |
| ] |
| our_table = [simple_table[0]] |
| for row in simple_table[1:]: |
| our_row = [row[0]] |
| for v in row[1:]: |
| our_row.append([v]) |
| our_table.append(our_row) |
| |
| tf = tabulator.TableFormatter(our_table, columns) |
| cell_table = tf.GetCellTable() |
| self.assertTrue(cell_table[0][0].colspan == 1) |
| self.assertTrue(cell_table[0][1].colspan == 2) |
| self.assertTrue(cell_table[0][2].colspan == 4) |
| self.assertTrue(cell_table[0][3].colspan == 4) |
| for row in cell_table[1:]: |
| for cell in row: |
| self.assertTrue(cell.colspan == 1) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |