Method test_sort

axlearn/cloud/common/utils_test.py:372–383

(self)

            table.get_col("c")

    def test_sort(self):
        table = utils.Table(headings=["a", "b"], rows=[[1, 4], [3, 2]])
        table.sort(key=lambda row: row[-1])
        self.assertEqual(
            utils.Table(headings=["a", "b"], rows=[[3, 2], [1, 4]]),
            table,
        )
        table.sort(key=lambda row: row[-1], reverse=True)
        self.assertEqual(
            utils.Table(headings=["a", "b"], rows=[[1, 4], [3, 2]]),
            table,
        )


class FlagConfigurableTest(parameterized.TestCase):