def test_cli_plots_tkagg(self, backend_mock, df_from_input_mock, show_mock):
backend_mock.return_value = 'macosx'
df_in = pd.DataFrame({
'x': range(1, 101),
'y': range(1, 101),
})
df_from_input_mock.return_value = df_in
sys.stdout = MagicMock()
main()
sys.stdout = sys.__stdout__
self.assertTrue(show_mock.called)
def test_cli_fit(self, df_from_input_mock, df_to_output_mock):
df_in = pd.DataFrame({
'x': range(1, 101),
'y': range(1, 101),
})
df_from_input_mock.return_value = df_in
main()
df_out = df_to_output_mock.call_args_list[0][0][1]
self.assertTrue(np.allclose(df_out.y, df_out.fit_))
self.assertTrue(np.allclose(df_out.y * 0, df_out.resid_))
def test_cli_fit(self, df_from_input_mock, df_to_output_mock):
df_in = pd.DataFrame({
'x': range(1, 101),
'y': range(1, 101),
})
df_from_input_mock.return_value = df_in
main()
df_out = df_to_output_mock.call_args_list[0][0][1]
self.assertTrue(np.allclose(df_out.y, df_out.fit_))
self.assertTrue(np.allclose(df_out.y * 0, df_out.resid_))
def test_cli_stats(self, df_from_input_mock, df_to_output_mock):
df_in = pd.DataFrame({
'x': range(1, 101),
'y': range(1, 101),
})
df_from_input_mock.return_value = df_in
write_mock = MagicMock()
sys.stdout = MagicMock()
sys.stdout.write = write_mock
main()
sys.stdout = sys.__stdout__
out_str = write_mock.call_args_list[0][0][0].replace('\n', ' ')
rex = re.compile(r'.*x\s+1\.0+')
m = rex.match(out_str)
self.assertTrue(True if m else False)
def test_cli_stats(self, df_from_input_mock, df_to_output_mock):
df_in = pd.DataFrame({
'x': range(1, 101),
'y': range(1, 101),
})
df_from_input_mock.return_value = df_in
write_mock = MagicMock()
sys.stdout = MagicMock()
sys.stdout.write = write_mock
main()
sys.stdout = sys.__stdout__
out_str = write_mock.call_args_list[0][0][0].replace('\n', ' ')
rex = re.compile(r'.*x\s+1\.0+')
m = rex.match(out_str)
self.assertTrue(True if m else False)
def test_cli_plots_osx(self, get_module_mock, df_from_input_mock):
backend_mock = MagicMock(lower=MagicMock(return_value='macosx'))
mpl_mock = MagicMock(get_backend=MagicMock(return_value=backend_mock))
show_mock = MagicMock()
plot_lib_mock = MagicMock(show=show_mock)
get_module_mock.side_effect = [
plot_lib_mock,
mpl_mock,
MagicMock(return_value=MagicMock()),
MagicMock(return_value=MagicMock()),
]
df_in = pd.DataFrame({
'x': range(1, 101),
'y': range(1, 101),
})
df_from_input_mock.return_value = df_in
sys.stdout = MagicMock()
main()
sys.stdout = sys.__stdout__
self.assertTrue(show_mock.called)
def test_cli_plots_osx(self, get_module_mock, df_from_input_mock):
backend_mock = MagicMock(lower=MagicMock(return_value='macosx'))
mpl_mock = MagicMock(get_backend=MagicMock(return_value=backend_mock))
show_mock = MagicMock()
plot_lib_mock = MagicMock(show=show_mock)
get_module_mock.side_effect = [
plot_lib_mock,
mpl_mock,
MagicMock(return_value=MagicMock()),
MagicMock(return_value=MagicMock()),
]
df_in = pd.DataFrame({
'x': range(1, 101),
'y': range(1, 101),
})
df_from_input_mock.return_value = df_in
sys.stdout = MagicMock()
main()
sys.stdout = sys.__stdout__
self.assertTrue(show_mock.called)
