zmc
2023-10-12 ed135d79df12a2466b52dae1a82326941211dcc9
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
"""
Tests for behavior if an author does *not* implement EA methods.
"""
import numpy as np
import pytest
 
from pandas.core.arrays import ExtensionArray
 
 
class MyEA(ExtensionArray):
    def __init__(self, values) -> None:
        self._values = values
 
 
@pytest.fixture
def data():
    arr = np.arange(10)
    return MyEA(arr)
 
 
class TestExtensionArray:
    def test_errors(self, data, all_arithmetic_operators):
        # invalid ops
        op_name = all_arithmetic_operators
        with pytest.raises(AttributeError):
            getattr(data, op_name)