# Copyright 2012 Google Inc. All Rights Reserved.

"""Tests for the tabulator module."""

__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()