feat(synset): add .size

medzuslovjansky · Oct 5, 2022 · f683788 · f683788
1 parent 1a6642f
commit f683788
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/core/Synset.test.ts b/src/core/Synset.test.ts
@@ -149,6 +149,18 @@ describe('Synset', () => {
     });
   });
 
+  describe('.size', () => {
+    it('should return lemma count', () => {
+      const synset = anEmptySynset();
+      expect(synset.size).toBe(0);
+      synset.add(new LemmaGroup());
+      synset.add(new LemmaGroup());
+      expect(synset.size).toBe(0);
+      synset.add(['some1', 'some2']);
+      expect(synset.size).toBe(2);
+    });
+  });
+
   describe('.union()', () => {
     it('should create a new synset with all the values', () => {
       const synset1 = anEmptySynset().add(['a', 'b']);

diff --git a/src/core/Synset.ts b/src/core/Synset.ts
@@ -108,6 +108,10 @@ export class Synset {
     return !!this.lemmas().next().done;
   }
 
+  public get size(): number {
+    return [...this.lemmas()].length;
+  }
+
   public union(
     other: Synset,
     equals: EqualityPredicate<Lemma> = valueEquals,