expose __len__ for IntervalSets

IMTMarburg · Mar 18, 2024 · 4db988b · 4db988b
1 parent ea8a808
commit 4db988b
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "mbf_nested_intervals"
-version = "0.3.1"
+version = "0.4.0"
 authors = ["Florian Finkernagel <finkernagel@imt.uni-marburg.de>"]
 edition = "2018"
 

diff --git a/mbf_nested_intervals/__init__.py b/mbf_nested_intervals/__init__.py
@@ -3,7 +3,7 @@
 import pandas as pd
 import itertools
 
-__version__ = "0.3.1"
+__version__ = "0.4.0"
 
 
 def _df_to_tup(df):

diff --git a/setup.cfg b/setup.cfg
@@ -1,7 +1,7 @@
 [metadata]
 name = mbf_nested_intervals
 description = wrapper around https://crates.io/crates/nested_intervals
-version = 0.3.1
+version = 0.4.0
 author = Florian Finkernagel
 author-email = finkernagel@imt.uni-marburg.de
 license = mit

diff --git a/src/lib.rs b/src/lib.rs
@@ -422,6 +422,10 @@ impl PyIntervalSet {
     }
 
 
+    pub fn __len__(&self) -> PyResult<usize> {
+        Ok(self.inner.len())
+    }
+
 }
 
 /// Wrapper around nested intervals

diff --git a/tests/test_interval_set.py b/tests/test_interval_set.py
@@ -14,6 +14,7 @@ def test_from_tuples(self):
             (1,15),
             (1,10),
         ]
+        assert len(i) == 3
 
     def test_from_tuples_with_id(self):
         i = IntervalSet.from_tuples_with_id([