Implement Fasta.__len__ for #157

mdshw5 · Jan 16, 2020 · a6bc9a8 · a6bc9a8
1 parent 43df193
commit a6bc9a8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pyfaidx/__init__.py b/pyfaidx/__init__.py
@@ -1020,6 +1020,10 @@ def __repr__(self):
     def __iter__(self):
         for rname in self.keys():
             yield self[rname]
+
+    def __len__(self):
+        """Return the cumulative length of all FastaRecords in self.records."""
+        return sum(len(record) for record in self)
 
     def get_seq(self, name, start, end, rc=False):
         """Return a sequence by record name and interval [start, end).