<s> from mock import patch , MagicMock <EOL> from unittest import TestCase <EOL> from pandashells . bin . p_plot import main <EOL> class MainTests ( TestCase ) : <EOL> @ patch ( '<STR_LIT>' ) <EOL> @ patch ( '<STR_LIT>' ) <EOL> @ patch ( '<STR_LIT>' ) <EOL> @ patch ( '<STR_LIT>' ) <EOL> @ patch ( '<STR_LIT>' ) <EOL> def test_plotting ( <EOL> self , draw_xy_mock , set_plot_styling_mock , df_from_input_mock , <EOL> add_args_mock , ArgumentParserMock ) : <EOL> args = MagicMock ( ) <EOL> parser = MagicMock ( parse_args = MagicMock ( return_value = args ) ) <EOL> ArgumentParserMock . return_value = parser <EOL> df_from_input_mock . return_value = '<STR_LIT>' <EOL> main ( ) <EOL> add_args_mock . assert_called_with ( <EOL> parser , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ) <EOL> df_from_input_mock . assert_called_with ( args ) <EOL> set_plot_styling_mock . assert_called_with ( args ) <EOL> draw_xy_mock . assert_called_with ( args , '<STR_LIT>' ) </s>
