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))