def test_StackedBytesArray_vectorized_indexing():
array = np.array([[b'a', b'b', b'c'], [b'd', b'e', b'f']], dtype='S')
stacked = strings.StackedBytesArray(array)
expected = np.array([[b'abc', b'def'], [b'def', b'abc']])
V = IndexerMaker(indexing.VectorizedIndexer)
indexer = V[np.array([[0, 1], [1, 0]])]
actual = stacked[indexer]
assert_array_equal(actual, expected)
def test_StackedBytesArray_vectorized_indexing():
array = np.array([[b"a", b"b", b"c"], [b"d", b"e", b"f"]], dtype="S")
stacked = strings.StackedBytesArray(array)
expected = np.array([[b"abc", b"def"], [b"def", b"abc"]])
V = IndexerMaker(indexing.VectorizedIndexer)
indexer = V[np.array([[0, 1], [1, 0]])]
actual = stacked[indexer]
assert_array_equal(actual, expected)
def test_StackedBytesArray():
array = np.array([[b'a', b'b', b'c'], [b'd', b'e', b'f']], dtype='S')
actual = strings.StackedBytesArray(array)
expected = np.array([b'abc', b'def'], dtype='S')
assert actual.dtype == expected.dtype
assert actual.shape == expected.shape
assert actual.size == expected.size
assert actual.ndim == expected.ndim
assert len(actual) == len(expected)
assert_array_equal(expected, actual)
B = IndexerMaker(indexing.BasicIndexer)
assert_array_equal(expected[:1], actual[B[:1]])
with pytest.raises(IndexError):
actual[B[:, :2]]
def test_StackedBytesArray():
array = np.array([[b"a", b"b", b"c"], [b"d", b"e", b"f"]], dtype="S")
actual = strings.StackedBytesArray(array)
expected = np.array([b"abc", b"def"], dtype="S")
assert actual.dtype == expected.dtype
assert actual.shape == expected.shape
assert actual.size == expected.size
assert actual.ndim == expected.ndim
assert len(actual) == len(expected)
assert_array_equal(expected, actual)
B = IndexerMaker(indexing.BasicIndexer)
assert_array_equal(expected[:1], actual[B[:1]])
with pytest.raises(IndexError):
actual[B[:, :2]]
def test_StackedBytesArray_scalar():
array = np.array([b'a', b'b', b'c'], dtype='S')
actual = strings.StackedBytesArray(array)
expected = np.array(b'abc')
assert actual.dtype == expected.dtype
assert actual.shape == expected.shape
assert actual.size == expected.size
assert actual.ndim == expected.ndim
with pytest.raises(TypeError):
len(actual)
np.testing.assert_array_equal(expected, actual)
B = IndexerMaker(indexing.BasicIndexer)
with pytest.raises(IndexError):
actual[B[:2]]
def test_StackedBytesArray_scalar() -> None:
array = np.array([b"a", b"b", b"c"], dtype="S")
actual = strings.StackedBytesArray(array)
expected = np.array(b"abc")
assert actual.dtype == expected.dtype
assert actual.shape == expected.shape
assert actual.size == expected.size
assert actual.ndim == expected.ndim
with pytest.raises(TypeError):
len(actual)
np.testing.assert_array_equal(expected, actual)
B = IndexerMaker(indexing.BasicIndexer)
with pytest.raises(IndexError):
actual[B[:2]]
