Text Extensions for Pandas: Tips and Techniques for Extending Pandas, Part 1 Blog

tep_extending_pandas_blog1_7.py

@pytest.fixture
def dtype():
    """ Return dtype of your extension array."""
    return TensorDtype()
 
@pytest.fixture
def data(dtype):
    """ Return an extension array as data for the tests."""
    return pd.array(np.array([[i] for i in range(100)]), dtype=dtype)
  
# Include other fixtures required by base tests.
 
class TestPandasInterface(base.BaseInterfaceTests):
    """ Extend the base Interface tests to run them all."""
    pass
  
# Add more test classes as needed.