예제 #1
0
 def test_apply_custom_formatting(self, mock_plt_savefig, mock_plt_show):
     viz.get_bias_plots(self.clf,
                        self.data,
                        self.features, ['under_30', 'is_female'],
                        s=10)
예제 #2
0
 def test_ignore_bad_plt_kwargs(self, mock_plt_savefig, mock_plt_show):
     viz.get_bias_plots(self.clf,
                        self.data,
                        self.features, ['under_30', 'is_female'],
                        s=10,
                        bad_kwarg=10)
예제 #3
0
 def test_use_default_values(self, mock_plt_savefig, mock_plt_show):
     viz.get_bias_plots(self.clf, self.data, self.features,
                        ['under_30', 'is_female'])