예제 #1
0
파일: testTable.py 프로젝트: pyGSTio/pyGSTi
    def test_general(self):

        table = ReportTable(['1.0'], ['Normal'])
        table.addrow(['1.0'], ['Normal'])
        table.render('text')

        self.assertTrue(table.has_key('1.0'))

        rowLabels = list(table.keys())
        row1Data  = table[rowLabels[0]]
        colLabels = list(row1Data.keys())

        self.assertTrue(rowLabels, table.row_names)
        self.assertTrue(colLabels, table.col_names)
        self.assertTrue(len(rowLabels), table.num_rows)
        self.assertTrue(len(colLabels), table.num_cols)

        el00 = table[rowLabels[0]][colLabels[0]]
        self.assertTrue( rowLabels[0] in table )
        self.assertTrue( rowLabels[0] in table )

        table_len = len(table)
        self.assertEqual(table_len, table.num_rows)

        table_as_str = str(table)
        row1a = table.row(key=rowLabels[0])
        col1a = table.col(key=colLabels[0])
        row1b = table.row(index=0)
        col1b = table.col(index=0)
        self.assertEqual(row1a,row1b)
        self.assertEqual(col1a,col1b)

        with self.assertRaises(KeyError):
            table['foobar']
        with self.assertRaises(KeyError):
            table.row(key='foobar') #invalid key
        with self.assertRaises(ValueError):
            table.row(index=100000) #out of bounds
        with self.assertRaises(ValueError):
            table.row() #must specify key or index
        with self.assertRaises(ValueError):
            table.row(key='foobar',index=1) #cannot specify key and index
        with self.assertRaises(KeyError):
            table.col(key='foobar') #invalid key
        with self.assertRaises(ValueError):
            table.col(index=100000) #out of bounds
        with self.assertRaises(ValueError):
            table.col() #must specify key or index
        with self.assertRaises(ValueError):
            table.col(key='foobar',index=1) #cannot specify key and index

        with self.assertRaises(ValueError):
            table.render(fmt="foobar") #invalid format
예제 #2
0
    def test_general(self):

        table = ReportTable(['1.0'], ['Normal'])
        table.addrow(['1.0'], ['Normal'])
        table.render('text')

        self.assertTrue(table.has_key('1.0'))

        rowLabels = list(table.keys())
        row1Data = table[rowLabels[0]]
        colLabels = list(row1Data.keys())

        self.assertTrue(rowLabels, table.row_names)
        self.assertTrue(colLabels, table.col_names)
        self.assertTrue(len(rowLabels), table.num_rows)
        self.assertTrue(len(colLabels), table.num_cols)

        el00 = table[rowLabels[0]][colLabels[0]]
        self.assertTrue(rowLabels[0] in table)
        self.assertTrue(rowLabels[0] in table)

        table_len = len(table)
        self.assertEqual(table_len, table.num_rows)

        table_as_str = str(table)
        row1a = table.row(key=rowLabels[0])
        col1a = table.col(key=colLabels[0])
        row1b = table.row(index=0)
        col1b = table.col(index=0)
        self.assertEqual(row1a, row1b)
        self.assertEqual(col1a, col1b)

        with self.assertRaises(KeyError):
            table['foobar']
        with self.assertRaises(KeyError):
            table.row(key='foobar')  #invalid key
        with self.assertRaises(ValueError):
            table.row(index=100000)  #out of bounds
        with self.assertRaises(ValueError):
            table.row()  #must specify key or index
        with self.assertRaises(ValueError):
            table.row(key='foobar', index=1)  #cannot specify key and index
        with self.assertRaises(KeyError):
            table.col(key='foobar')  #invalid key
        with self.assertRaises(ValueError):
            table.col(index=100000)  #out of bounds
        with self.assertRaises(ValueError):
            table.col()  #must specify key or index
        with self.assertRaises(ValueError):
            table.col(key='foobar', index=1)  #cannot specify key and index

        with self.assertRaises(ValueError):
            table.render(fmt="foobar")  #invalid format
예제 #3
0
class TableInstanceTester(BaseCase):
    custom_headings = {'html': 'test', 'python': 'test', 'latex': 'test'}

    def setUp(self):
        self.table = ReportTable(self.custom_headings,
                                 ['Normal'] * 4)  # Four formats

    def test_element_accessors(self):
        self.table.add_row(['1.0'], ['Normal'])

        self.assertTrue('1.0' in self.table)

        self.assertEqual(len(self.table), self.table.num_rows)

        row_by_key = self.table.row(key=self.table.row_names[0])
        row_by_idx = self.table.row(index=0)
        self.assertEqual(row_by_key, row_by_idx)

        col_by_key = self.table.col(key=self.table.col_names[0])
        col_by_idx = self.table.col(index=0)
        self.assertEqual(col_by_key, col_by_idx)

    def test_to_string(self):
        s = str(self.table)
        # TODO assert correctness

    def test_render_HTML(self):
        self.table.add_row(['1.0'], ['Normal'])
        self.table.add_row(['1.0'], ['Normal'])
        render = self.table.render('html')
        # TODO assert correctness

    def test_render_LaTeX(self):
        self.table.add_row(['1.0'], ['Normal'])
        self.table.add_row(['1.0'], ['Normal'])
        render = self.table.render('latex')
        # TODO assert correctness

    def test_finish(self):
        self.table.add_row(['1.0'], ['Normal'])
        self.table.finish()
        # TODO assert correctness

    def test_render_raises_on_unknown_format(self):
        with self.assertRaises(NotImplementedError):
            self.table.render('foobar')

    def test_raise_on_invalid_accessor(self):
        # XXX are these neccessary?  EGN: maybe not - checks invalid inputs, which maybe shouldn't need testing?
        with self.assertRaises(KeyError):
            self.table['foobar']
        with self.assertRaises(KeyError):
            self.table.row(key='foobar')  # invalid key
        with self.assertRaises(ValueError):
            self.table.row(index=100000)  # out of bounds
        with self.assertRaises(ValueError):
            self.table.row()  # must specify key or index
        with self.assertRaises(ValueError):
            self.table.row(key='foobar',
                           index=1)  # cannot specify key and index
        with self.assertRaises(KeyError):
            self.table.col(key='foobar')  # invalid key
        with self.assertRaises(ValueError):
            self.table.col(index=100000)  # out of bounds
        with self.assertRaises(ValueError):
            self.table.col()  # must specify key or index
        with self.assertRaises(ValueError):
            self.table.col(key='foobar',
                           index=1)  # cannot specify key and index