def test_number_to_dataframe(self, sys_mock):
cmd = ''
df = 7
sys_mock.stderr = MagicMock(write=MagicMock())
sys_mock.exit = MagicMock()
framify(cmd, df)
self.assertTrue(sys_mock.stderr.write.called)
self.assertTrue(sys_mock.exit.called)
def test_series_to_dataframe(self):
cmd = ''
df = pd.Series({'a': 1})
out = framify(cmd, df)
self.assertTrue(isinstance(out, pd.DataFrame))
def test_list_to_dataframe(self):
cmd = ''
df = [1, 2, 3]
out = framify(cmd, df)
self.assertTrue(isinstance(out, pd.DataFrame))
def test_dataframe_to_dataframe(self):
cmd = ''
df = pd.DataFrame([{'a': 1}])
out = framify(cmd, df)
self.assertTrue(isinstance(out, pd.DataFrame))
