Exemplos de _load_data em Python

Exemplo n.º 1

    def test_load_data_path(self, tmpdir):
        path = os.path.join(tmpdir, 'data.csv')
        self.data.to_csv(path, index=False)

        data = functional._load_data(path)

        pd.testing.assert_frame_equal(data, self.data)

Exemplo n.º 2

    def test_load_data_none(self):
        data = functional._load_data(None)

        assert data is None

Exemplo n.º 3

    def test_load_data_df(self):
        data = functional._load_data(self.data)

        assert data is self.data