Add __contains__ to MinMax

asmeurer · asmeurer · commit 94659590b8db · 2024-07-03T13:04:58.000-06:00
diff --git a/array_api_tests/dtype_helpers.py b/array_api_tests/dtype_helpers.py
@@ -209,6 +209,9 @@ class MinMax(NamedTuple):
     min: Union[int, float]
     max: Union[int, float]
 
+    def __contains__(self, other):
+        assert isinstance(other, (int, float))
+        return self.min <= other <= self.max
 
 dtype_ranges = _make_dtype_mapping_from_names(
     {

Original file line number	Diff line number	Diff line change
`@@ -209,6 +209,9 @@ class MinMax(NamedTuple):`
`209`	`209`	`min: Union[int, float]`
`210`	`210`	`max: Union[int, float]`
`211`	`211`
	`212`	`+ def __contains__(self, other):`
	`213`	`+ assert isinstance(other, (int, float))`
	`214`	`+ return self.min <= other <= self.max`
`212`	`215`
`213`	`216`	`dtype_ranges = _make_dtype_mapping_from_names(`
`214`	`217`	`{`