Skip to content

Commit 3be7d25

Browse files
committed
Add Vector.str.contains shorthand
1 parent 87e701c commit 3be7d25

File tree

3 files changed

+11
-0
lines changed

3 files changed

+11
-0
lines changed

NEWS.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,8 @@
1+
PENDING: Dataiter 1.3
2+
=====================
3+
4+
* `Vector.str.contains`: New proxy method (equivalent to find(x) > -1)
5+
16
2025-10-05: Dataiter 1.2
27
========================
38

dataiter/test/test_vector.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -508,6 +508,11 @@ def test_str(self):
508508
assert a.str.str_len().tolist() == [4, 4]
509509
assert a.str.upper().tolist() == ["ASDF", "1234"]
510510

511+
def test_str_contains(self):
512+
a = Vector(["asdf", "1234"])
513+
assert a.str.contains("df").tolist() == [True, False]
514+
assert a.str.contains("23").tolist() == [False, True]
515+
511516
def test_tail(self):
512517
a = Vector([1, 2, 3, 4, 5])
513518
assert a.tail(3).tolist() == [3, 4, 5]

dataiter/vector.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,7 @@ def __init__(self, vector):
8888
self.add = wrap("add")
8989
self.capitalize = wrap("capitalize")
9090
self.center = wrap("center")
91+
self.contains = lambda x: self.find(x) > -1
9192
self.count = wrap("count")
9293
self.decode = wrap("decode")
9394
self.encode = wrap("encode")

0 commit comments

Comments
 (0)