141 lines
4.3 KiB
Python
141 lines
4.3 KiB
Python
# 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):
|
|
"""Tests for the Tabulator class."""
|
|
|
|
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
|
|
# pylint: disable=protected-access
|
|
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()
|