Inzynierka_Gwiazdy/machine_learning/Lib/site-packages/pandas/tests/apply/conftest.py
2023-09-20 19:46:58 +02:00

19 lines
399 B
Python

import numpy as np
import pytest
from pandas import DataFrame
@pytest.fixture
def int_frame_const_col():
"""
Fixture for DataFrame of ints which are constant per column
Columns are ['A', 'B', 'C'], with values (per column): [1, 2, 3]
"""
df = DataFrame(
np.tile(np.arange(3, dtype="int64"), 6).reshape(6, -1) + 1,
columns=["A", "B", "C"],
)
return df