Python autofill_plot_fields_and_labelsの例

コード例 #1

ファイル: plot_lib_tests.py プロジェクト: richwu/pandashells

    def test_autofill_plot_fields_and_labels_do_nothing(self):
        """autofill_plot_fields_and_labels does no filling
        """
        args = MagicMock(x=None, xlabel="xpre", ylabel="ypre")
        df = MagicMock(columns=[1])

        plot_lib.autofill_plot_fields_and_labels(args, df)
        self.assertEqual(args.xlabel, "xpre")
        self.assertEqual(args.ylabel, "ypre")

コード例 #2

    def test_autofill_plot_fields_and_labels_do_nothing(self):
        """autofill_plot_fields_and_labels does no filling
        """
        args = MagicMock(x=None, xlabel='xpre', ylabel='ypre')
        df = MagicMock(columns=[1])

        plot_lib.autofill_plot_fields_and_labels(args, df)
        self.assertEqual(args.xlabel, 'xpre')
        self.assertEqual(args.ylabel, 'ypre')

コード例 #3

ファイル: plot_lib_tests.py プロジェクト: richwu/pandashells

    def test_autofill_plot_fields_and_labels_2_cols(self):
        """autofill_plot_labels() appropriately handles 2 column frame
        """
        args = MagicMock(x=None, xlabel=None, ylabel=None)
        df = MagicMock(columns=["x", "y"])

        plot_lib.autofill_plot_fields_and_labels(args, df)
        self.assertEqual(args.x, ["x"])
        self.assertEqual(args.y, ["y"])
        self.assertEqual(args.xlabel, ["x"])
        self.assertEqual(args.ylabel, ["y"])

コード例 #4

    def test_autofill_plot_fields_and_labels_2_cols(self):
        """autofill_plot_labels() appropriately handles 2 column frame
        """
        args = MagicMock(x=None, xlabel=None, ylabel=None)
        df = MagicMock(columns=['x', 'y'])

        plot_lib.autofill_plot_fields_and_labels(args, df)
        self.assertEqual(args.x, ['x'])
        self.assertEqual(args.y, ['y'])
        self.assertEqual(args.xlabel, ['x'])
        self.assertEqual(args.ylabel, ['y'])