bee_python 0.0.3 → 0.0.4

data/tools/logilab/logilab/common/test/unittest_table.py

@@ -1,448 +0,0 @@
-# copyright 2003-2010 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
-# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
-#
-# This file is part of logilab-common.
-#
-# logilab-common is free software: you can redistribute it and/or modify it under
-# the terms of the GNU Lesser General Public License as published by the Free
-# Software Foundation, either version 2.1 of the License, or (at your option) any
-# later version.
-#
-# logilab-common is distributed in the hope that it will be useful, but WITHOUT
-# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
-# FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
-# details.
-#
-# You should have received a copy of the GNU Lesser General Public License along
-# with logilab-common.  If not, see <http://www.gnu.org/licenses/>.
-"""
-Unittests for table management
-"""
-
-__revision__ = '$Id: unittest_table.py,v 1.1 2010/06/14 14:06:58 casa Exp $'
-
-import sys
-import os
-from cStringIO import StringIO
-
-from logilab.common.testlib import TestCase, unittest_main
-from logilab.common.table import Table, TableStyleSheet, DocbookTableWriter, \
-     DocbookRenderer, TableStyle, TableWriter, TableCellRenderer
-from logilab.common.compat import set
-
-class TableTC(TestCase):
-    """Table TestCase class"""
-
-    def setUp(self):
-        """Creates a default table"""
-        # from logilab.common import table
-        # reload(table)
-        self.table = Table()
-        self.table.create_rows(['row1', 'row2', 'row3'])
-        self.table.create_columns(['col1', 'col2'])
-
-    def test_valeur_scalaire(self):
-        tab = Table()
-        tab.create_columns(['col1'])
-        tab.append_row([1])
-        self.assertEquals(tab, [[1]])
-        tab.append_row([2])
-        self.assertEquals(tab[0,0], 1)
-        self.assertEquals(tab[1,0], 2)
-
-    def test_valeur_ligne(self):
-        tab = Table()
-        tab.create_columns(['col1','col2'])
-        tab.append_row([1,2])
-        self.assertEquals(tab, [[1,2]])
-
-    def test_valeur_colonne(self):
-        tab = Table()
-        tab.create_columns(['col1'])
-        tab.append_row([1])
-        tab.append_row([2])
-        self.assertEquals(tab, [[1],[2]])
-        self.assertEquals(tab[:,0], [1,2])
-
-    def test_indexation(self):
-        """we should be able to use [] to access rows"""
-        self.assert_(self.table[0] == self.table.data[0])
-        self.assert_(self.table[1] == self.table.data[1])
-
-    def test_iterable(self):
-        """test iter(table)"""
-        it = iter(self.table)
-        self.assert_(it.next() == self.table.data[0])
-        self.assert_(it.next() == self.table.data[1])
-
-    def test_get_rows(self):
-        """tests Table.get_rows()"""
-        self.assertEquals(self.table, [[0, 0], [0, 0], [0, 0]])
-        self.assertEquals(self.table[:], [[0, 0], [0, 0], [0, 0]])
-        self.table.insert_column(1, range(3), 'supp')
-        self.assertEquals(self.table, [[0, 0, 0], [0, 1, 0], [0, 2, 0]])
-        self.assertEquals(self.table[:], [[0, 0, 0], [0, 1, 0], [0, 2, 0]])
-
-    def test_get_cells(self):
-        self.table.insert_column(1, range(3), 'supp')
-        self.assertEquals(self.table[0,1], 0)
-        self.assertEquals(self.table[1,1], 1)
-        self.assertEquals(self.table[2,1], 2)
-        self.assertEquals(self.table['row1', 'supp'], 0)
-        self.assertEquals(self.table['row2', 'supp'], 1)
-        self.assertEquals(self.table['row3', 'supp'], 2)
-        self.assertRaises(KeyError, self.table.__getitem__, ('row1', 'foo'))
-        self.assertRaises(KeyError, self.table.__getitem__, ('foo', 'bar'))
-
-    def test_shape(self):
-        """tests table shape"""
-        self.assertEquals(self.table.shape, (3, 2))
-        self.table.insert_column(1, range(3), 'supp')
-        self.assertEquals(self.table.shape, (3, 3))
-
-    def test_set_column(self):
-        """Tests that table.set_column() works fine.
-        """
-        self.table.set_column(0, range(3))
-        self.assertEquals(self.table[0,0], 0)
-        self.assertEquals(self.table[1,0], 1)
-        self.assertEquals(self.table[2,0], 2)
-
-    def test_set_column_by_id(self):
-        """Tests that table.set_column_by_id() works fine.
-        """
-        self.table.set_column_by_id('col1', range(3))
-        self.assertEquals(self.table[0,0], 0)
-        self.assertEquals(self.table[1,0], 1)
-        self.assertEquals(self.table[2,0], 2)
-        self.assertRaises(KeyError, self.table.set_column_by_id, 'col123', range(3))
-
-    def test_cells_ids(self):
-        """tests that we can access cells by giving row/col ids"""
-        self.assertRaises(KeyError, self.table.set_cell_by_ids, 'row12', 'col1', 12)
-        self.assertRaises(KeyError, self.table.set_cell_by_ids, 'row1', 'col12', 12)
-        self.assertEquals(self.table[0,0], 0)
-        self.table.set_cell_by_ids('row1', 'col1', 'DATA')
-        self.assertEquals(self.table[0,0], 'DATA')
-        self.assertRaises(KeyError, self.table.set_row_by_id, 'row12', [])
-        self.table.set_row_by_id('row1', ['1.0', '1.1'])
-        self.assertEquals(self.table[0,0], '1.0')
-
-    def test_insert_row(self):
-        """tests a row insertion"""
-        tmp_data = ['tmp1', 'tmp2']
-        self.table.insert_row(1, tmp_data, 'tmprow')
-        self.assertEquals(self.table[1], tmp_data)
-        self.assertEquals(self.table['tmprow'], tmp_data)
-        self.table.delete_row_by_id('tmprow')
-        self.assertRaises(KeyError, self.table.delete_row_by_id, 'tmprow')
-        self.assertEquals(self.table[1], [0, 0])
-        self.assertRaises(KeyError, self.table.__getitem__, 'tmprow')
-
-    def test_get_column(self):
-        """Tests that table.get_column() works fine.
-        """
-        self.table.set_cell(0, 1, 12)
-        self.table.set_cell(2, 1, 13)
-        self.assertEquals(self.table[:,1], [12,0,13])
-        self.assertEquals(self.table[:,'col2'], [12,0,13])
-
-    def test_get_columns(self):
-        """Tests if table.get_columns() works fine.
-        """
-        self.table.set_cell(0, 1, 12)
-        self.table.set_cell(2, 1, 13)
-        self.assertEquals(self.table.get_columns(), [[0,0,0], [12,0,13]])
-
-    def test_insert_column(self):
-        """Tests that table.insert_column() works fine.
-        """
-        self.table.insert_column(1, range(3), "inserted_column")
-        self.assertEquals(self.table[:,1], [0,1,2])
-        self.assertEquals(self.table.col_names,
-                          ['col1', 'inserted_column', 'col2'])
-
-    def test_delete_column(self):
-        """Tests that table.delete_column() works fine.
-        """
-        self.table.delete_column(1)
-        self.assertEquals(self.table.col_names, ['col1'])
-        self.assertEquals(self.table[:,0], [0,0,0])
-        self.assertRaises(KeyError, self.table.delete_column_by_id, 'col2')
-        self.table.delete_column_by_id('col1')
-        self.assertEquals(self.table.col_names, [])
-
-    def test_transpose(self):
-        """Tests that table.transpose() works fine.
-        """
-        self.table.append_column(range(5,8), 'col3')
-        ttable = self.table.transpose()
-        self.assertEquals(ttable.row_names, ['col1', 'col2', 'col3'])
-        self.assertEquals(ttable.col_names, ['row1', 'row2', 'row3'])
-        self.assertEquals(ttable.data, [[0,0,0], [0,0,0], [5,6,7]])
-
-    def test_sort_table(self):
-        """Tests the table sort by column
-        """
-        self.table.set_column(0, [3, 1, 2])
-        self.table.set_column(1, [1, 2, 3])
-        self.table.sort_by_column_index(0)
-        self.assertEquals(self.table.row_names, ['row2', 'row3', 'row1'])
-        self.assertEquals(self.table.data, [[1, 2], [2, 3], [3, 1]])
-        self.table.sort_by_column_index(1, 'desc')
-        self.assertEquals(self.table.row_names, ['row3', 'row2', 'row1'])
-        self.assertEquals(self.table.data, [[2, 3], [1, 2], [3, 1]])
-
-    def test_sort_by_id(self):
-        """tests sort_by_column_id()"""
-        self.table.set_column_by_id('col1', [3, 1, 2])
-        self.table.set_column_by_id('col2', [1, 2, 3])
-        self.table.sort_by_column_id('col1')
-        self.assertRaises(KeyError, self.table.sort_by_column_id, 'col123')
-        self.assertEquals(self.table.row_names, ['row2', 'row3', 'row1'])
-        self.assertEquals(self.table.data, [[1, 2], [2, 3], [3, 1]])
-        self.table.sort_by_column_id('col2', 'desc')
-        self.assertEquals(self.table.row_names, ['row3', 'row2', 'row1'])
-        self.assertEquals(self.table.data, [[2, 3], [1, 2], [3, 1]])
-
-    def test_pprint(self):
-        """only tests pprint doesn't raise an exception"""
-        self.table.pprint()
-        str(self.table)
-
-
-class GroupByTC(TestCase):
-    """specific test suite for groupby()"""
-    def setUp(self):
-        t = Table()
-        t.create_columns(['date', 'res', 'task', 'usage'])
-        t.append_row(['date1', 'ing1', 'task1', 0.3])
-        t.append_row(['date1', 'ing2', 'task2', 0.3])
-        t.append_row(['date2', 'ing3', 'task3', 0.3])
-        t.append_row(['date3', 'ing4', 'task2', 0.3])
-        t.append_row(['date1', 'ing1', 'task3', 0.3])
-        t.append_row(['date3', 'ing1', 'task3', 0.3])
-        self.table = t
-
-    def test_single_groupby(self):
-        """tests groupby() on several columns"""
-        grouped = self.table.groupby('date')
-        self.assertEquals(len(grouped), 3)
-        self.assertEquals(len(grouped['date1']), 3)
-        self.assertEquals(len(grouped['date2']), 1)
-        self.assertEquals(len(grouped['date3']), 2)
-        self.assertEquals(grouped['date1'], [
-            ('date1', 'ing1', 'task1', 0.3),
-            ('date1', 'ing2', 'task2', 0.3),
-            ('date1', 'ing1', 'task3', 0.3),
-            ])
-        self.assertEquals(grouped['date2'], [('date2', 'ing3', 'task3', 0.3)])
-        self.assertEquals(grouped['date3'], [
-            ('date3', 'ing4', 'task2', 0.3),
-            ('date3', 'ing1', 'task3', 0.3),
-            ])
-
-    def test_multiple_groupby(self):
-        """tests groupby() on several columns"""
-        grouped = self.table.groupby('date', 'task')
-        self.assertEquals(len(grouped), 3)
-        self.assertEquals(len(grouped['date1']), 3)
-        self.assertEquals(len(grouped['date2']), 1)
-        self.assertEquals(len(grouped['date3']), 2)
-        self.assertEquals(grouped['date1']['task1'], [('date1', 'ing1', 'task1', 0.3)])
-        self.assertEquals(grouped['date2']['task3'], [('date2', 'ing3', 'task3', 0.3)])
-        self.assertEquals(grouped['date3']['task2'], [('date3', 'ing4', 'task2', 0.3)])
-        date3 = grouped['date3']
-        self.assertRaises(KeyError, date3.__getitem__, 'task1')
-
-
-    def test_select(self):
-        """tests Table.select() method"""
-        rows = self.table.select('date', 'date1')
-        self.assertEquals(rows, [
-            ('date1', 'ing1', 'task1', 0.3),
-            ('date1', 'ing2', 'task2', 0.3),
-            ('date1', 'ing1', 'task3', 0.3),
-            ])
-
-class TableStyleSheetTC(TestCase):
-    """The Stylesheet test case
-    """
-    def setUp(self):
-        """Builds a simple table to test the stylesheet
-        """
-        self.table = Table()
-        self.table.create_row('row1')
-        self.table.create_columns(['a','b','c'])
-        self.stylesheet = TableStyleSheet()
-        # We don't want anything to be printed
-        self.stdout_backup = sys.stdout
-        sys.stdout = StringIO()
-
-    def tearDown(self):
-        sys.stdout = self.stdout_backup
-
-    def test_add_rule(self):
-        """Tests that the regex pattern works as expected.
-        """
-        rule = '0_2 = sqrt(0_0**2 + 0_1**2)'
-        self.stylesheet.add_rule(rule)
-        self.table.set_row(0, [3,4,0])
-        self.table.apply_stylesheet(self.stylesheet)
-        self.assertEquals(self.table[0], [3,4,5])
-        self.assertEquals(len(self.stylesheet.rules), 1)
-        self.stylesheet.add_rule('some bad rule with bad syntax')
-        self.assertEquals(len(self.stylesheet.rules), 1, "Ill-formed rule mustn't be added")
-        self.assertEquals(len(self.stylesheet.instructions), 1, "Ill-formed rule mustn't be added")
-
-    def test_stylesheet_init(self):
-        """tests Stylesheet.__init__"""
-        rule = '0_2 = 1'
-        sheet = TableStyleSheet([rule, 'bad rule'])
-        self.assertEquals(len(sheet.rules), 1, "Ill-formed rule mustn't be added")
-        self.assertEquals(len(sheet.instructions), 1, "Ill-formed rule mustn't be added")
-
-    def test_rowavg_rule(self):
-        """Tests that add_rowavg_rule works as expected
-        """
-        self.table.set_row(0, [10,20,0])
-        self.stylesheet.add_rowavg_rule((0,2), 0, 0, 1)
-        self.table.apply_stylesheet(self.stylesheet)
-        val = self.table[0,2]
-        self.assert_(int(val) == 15)
-
-
-    def test_rowsum_rule(self):
-        """Tests that add_rowsum_rule works as expected
-        """
-        self.table.set_row(0, [10,20,0])
-        self.stylesheet.add_rowsum_rule((0,2), 0, 0, 1)
-        self.table.apply_stylesheet(self.stylesheet)
-        val = self.table[0,2]
-        self.assert_(val == 30)
-
-
-    def test_colavg_rule(self):
-        """Tests that add_colavg_rule works as expected
-        """
-        self.table.set_row(0, [10,20,0])
-        self.table.append_row([12,8,3], 'row2')
-        self.table.create_row('row3')
-        self.stylesheet.add_colavg_rule((2,0), 0, 0, 1)
-        self.table.apply_stylesheet(self.stylesheet)
-        val = self.table[2,0]
-        self.assert_(int(val) == 11)
-
-
-    def test_colsum_rule(self):
-        """Tests that add_colsum_rule works as expected
-        """
-        self.table.set_row(0, [10,20,0])
-        self.table.append_row([12,8,3], 'row2')
-        self.table.create_row('row3')
-        self.stylesheet.add_colsum_rule((2,0), 0, 0, 1)
-        self.table.apply_stylesheet(self.stylesheet)
-        val = self.table[2,0]
-        self.assert_(val == 22)
-
-
-
-class TableStyleTC(TestCase):
-    """Test suite for TableSuite"""
-    def setUp(self):
-        self.table = Table()
-        self.table.create_rows(['row1', 'row2', 'row3'])
-        self.table.create_columns(['col1', 'col2'])
-        self.style = TableStyle(self.table)
-        self._tested_attrs = (('size', '1*'),
-                              ('alignment', 'right'),
-                              ('unit', ''))
-
-    def test_getset(self):
-        """tests style's get and set methods"""
-        for attrname, default_value in self._tested_attrs:
-            getter = getattr(self.style, 'get_%s' % attrname)
-            setter = getattr(self.style, 'set_%s' % attrname)
-            self.assertRaises(KeyError, getter, 'badcol')
-            self.assertEquals(getter('col1'), default_value)
-            setter('FOO', 'col1')
-            self.assertEquals(getter('col1'), 'FOO')
-
-    def test_getset_index(self):
-        """tests style's get and set by index methods"""
-        for attrname, default_value in self._tested_attrs:
-            getter = getattr(self.style, 'get_%s' % attrname)
-            setter = getattr(self.style, 'set_%s' % attrname)
-            igetter = getattr(self.style, 'get_%s_by_index' % attrname)
-            isetter = getattr(self.style, 'set_%s_by_index' % attrname)
-            self.assertEquals(getter('__row_column__'), default_value)
-            isetter('FOO', 0)
-            self.assertEquals(getter('__row_column__'), 'FOO')
-            self.assertEquals(igetter(0), 'FOO')
-            self.assertEquals(getter('col1'), default_value)
-            isetter('FOO', 1)
-            self.assertEquals(getter('col1'), 'FOO')
-            self.assertEquals(igetter(1), 'FOO')
-
-
-class RendererTC(TestCase):
-    """Test suite for DocbookRenderer"""
-    def setUp(self):
-        self.renderer = DocbookRenderer(alignment = True)
-        self.table = Table()
-        self.table.create_rows(['row1', 'row2', 'row3'])
-        self.table.create_columns(['col1', 'col2'])
-        self.style = TableStyle(self.table)
-        self.base_renderer = TableCellRenderer()
-
-    def test_cell_content(self):
-        """test how alignment is rendered"""
-        entry_xml = self.renderer._render_cell_content('data', self.style, 1)
-        self.assertEquals(entry_xml, "<entry align='right'>data</entry>\n")
-        self.style.set_alignment_by_index('left', 1)
-        entry_xml = self.renderer._render_cell_content('data', self.style, 1)
-        self.assertEquals(entry_xml, "<entry align='left'>data</entry>\n")
-
-    def test_default_content_rendering(self):
-        """tests that default rendering just prints the cell's content"""
-        rendered_cell = self.base_renderer._render_cell_content('data', self.style, 1)
-        self.assertEquals(rendered_cell, "data")
-
-    def test_replacement_char(self):
-        """tests that 0 is replaced when asked for"""
-        cell_content = self.base_renderer._make_cell_content(0, self.style, 1)
-        self.assertEquals(cell_content, 0)
-        self.base_renderer.properties['skip_zero'] = '---'
-        cell_content = self.base_renderer._make_cell_content(0, self.style, 1)
-        self.assertEquals(cell_content, '---')
-
-    def test_unit(self):
-        """tests if units are added"""
-        self.base_renderer.properties['units'] = True
-        self.style.set_unit_by_index('EUR', 1)
-        cell_content = self.base_renderer._make_cell_content(12, self.style, 1)
-        self.assertEquals(cell_content, '12 EUR')
-
-
-class DocbookTableWriterTC(TestCase):
-    """TestCase for table's writer"""
-    def setUp(self):
-        self.stream = StringIO()
-        self.table = Table()
-        self.table.create_rows(['row1', 'row2', 'row3'])
-        self.table.create_columns(['col1', 'col2'])
-        self.writer = DocbookTableWriter(self.stream, self.table, None)
-        self.writer.set_renderer(DocbookRenderer())
-
-    def test_write_table(self):
-        """make sure write_table() doesn't raise any exception"""
-        self.writer.write_table()
-
-    def test_abstract_writer(self):
-        """tests that Abstract Writers can't be used !"""
-        writer = TableWriter(self.stream, self.table, None)
-        self.assertRaises(NotImplementedError, writer.write_table)
-
-
-if __name__ == '__main__':
-    unittest_main()

data/tools/logilab/logilab/common/test/unittest_taskqueue.py

@@ -1,71 +0,0 @@
-# copyright 2003-2010 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
-# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
-#
-# This file is part of logilab-common.
-#
-# logilab-common is free software: you can redistribute it and/or modify it under
-# the terms of the GNU Lesser General Public License as published by the Free
-# Software Foundation, either version 2.1 of the License, or (at your option) any
-# later version.
-#
-# logilab-common is distributed in the hope that it will be useful, but WITHOUT
-# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
-# FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
-# details.
-#
-# You should have received a copy of the GNU Lesser General Public License along
-# with logilab-common.  If not, see <http://www.gnu.org/licenses/>.
-from logilab.common.testlib import TestCase, unittest_main
-
-from logilab.common.tasksqueue import *
-
-class TaskTC(TestCase):
-
-    def test_eq(self):
-        self.failIf(Task('t1') == Task('t2'))
-        self.failUnless(Task('t1') == Task('t1'))
-
-    def test_cmp(self):
-        self.failUnless(Task('t1', LOW) < Task('t2', MEDIUM))
-        self.failIf(Task('t1', LOW) > Task('t2', MEDIUM))
-        self.failUnless(Task('t1', HIGH) > Task('t2', MEDIUM))
-        self.failIf(Task('t1', HIGH) < Task('t2', MEDIUM))
-
-
-class PrioritizedTasksQueueTC(TestCase):
-
-    def test_priority(self):
-        queue = PrioritizedTasksQueue()
-        queue.put(Task('t1'))
-        queue.put(Task('t2', MEDIUM))
-        queue.put(Task('t3', HIGH))
-        queue.put(Task('t4', LOW))
-        self.assertEquals(queue.get().id, 't3')
-        self.assertEquals(queue.get().id, 't2')
-        self.assertEquals(queue.get().id, 't1')
-        self.assertEquals(queue.get().id, 't4')
-
-    def test_remove_equivalent(self):
-        queue = PrioritizedTasksQueue()
-        queue.put(Task('t1'))
-        queue.put(Task('t2', MEDIUM))
-        queue.put(Task('t1', HIGH))
-        queue.put(Task('t3', MEDIUM))
-        queue.put(Task('t2', MEDIUM))
-        self.assertEquals(queue.qsize(), 3)
-        self.assertEquals(queue.get().id, 't1')
-        self.assertEquals(queue.get().id, 't2')
-        self.assertEquals(queue.get().id, 't3')
-        self.assertEquals(queue.qsize(), 0)
-
-    def test_remove(self):
-        queue = PrioritizedTasksQueue()
-        queue.put(Task('t1'))
-        queue.put(Task('t2'))
-        queue.put(Task('t3'))
-        queue.remove('t2')
-        self.assertEquals([t.id for t in queue], ['t3', 't1'])
-        self.assertRaises(ValueError, queue.remove, 't4')
-
-if __name__ == '__main__':
-    unittest_main()