cov

hdmf-dev · Oct 23, 2023 · 02f8c11 · 02f8c11
1 parent 927eab8
commit 02f8c11
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 7 deletions.
diff --git a/src/hdmf/term_set.py b/src/hdmf/term_set.py
@@ -62,11 +62,11 @@ def __repr__(self):
         re = "Schema Path: %s\n" % self.term_schema_path
         re += "Sources: " + ", ".join(list(self.sources.keys()))+"\n"
         re += "Terms: \n"
-        i = 0
+        i = 1
         for term in terms:
             if i>4:
                 break
-            elif i==2:
+            elif i==3:
                 re += "   ... ... \n"
             re += "   - %s\n" % term
             i += 1
@@ -79,11 +79,11 @@ def _repr_html_(self):
         re = "<b>" + "Schema Path: " + "</b>" + self.term_schema_path + "<br>"
         re += "<b>" + "Sources: " + "</b>" + ", ".join(list(self.sources.keys())) + "<br>"
         re += "<b> Terms: </b>"
-        i = 0
+        i = 1
         for term in terms:
             if i>4:
                 break
-            elif i==2:
+            elif i==3:
                 re += "... ..."
             re += "<li> %s </li>" % term
             i += 1

diff --git a/tests/unit/example_test_term_set.yaml b/tests/unit/example_test_term_set.yaml
@@ -22,3 +22,6 @@ enums:
       Myrmecophaga tridactyla:
         description: the species is an anteater
         meaning: NCBI_TAXON:71006
+      Panda:
+        description: the species is panda
+        meaning: NCBI_TAXON:212257
diff --git a/tests/unit/test_term_set.py b/tests/unit/test_term_set.py
@@ -33,20 +33,21 @@ def test_repr(self):
         termset = TermSet(term_schema_path='tests/unit/example_test_term_set.yaml')
         output = ('Schema Path: tests/unit/example_test_term_set.yaml\nSources: NCBI_TAXON\nTerms: \n'+
                   '   - Homo sapiens\n   - Mus musculus\n   ... ... \n   - Ursus arctos horribilis\n'+
-                  '   - Myrmecophaga tridactyla\nNumber of terms: 4')
+                  '   - Myrmecophaga tridactyla\nNumber of terms: 5')
         self.assertEqual(repr(termset), output)
 
     def test_repr_html(self):
         termset = TermSet(term_schema_path='tests/unit/example_test_term_set.yaml')
         output = ('<b>Schema Path: </b>tests/unit/example_test_term_set.yaml<br><b>Sources:'+
                   ' </b>NCBI_TAXON<br><b> Terms: </b><li> Homo sapiens </li><li> Mus musculus'+
                   ' </li>... ...<li> Ursus arctos horribilis </li><li> Myrmecophaga tridactyla'+
-                  ' </li><i> Number of terms:</i> 4')
+                  ' </li><i> Number of terms:</i> 5')
         self.assertEqual(termset._repr_html_(), output)
 
     def test_view_set(self):
         termset = TermSet(term_schema_path='tests/unit/example_test_term_set.yaml')
-        expected = ['Homo sapiens', 'Mus musculus', 'Ursus arctos horribilis', 'Myrmecophaga tridactyla']
+        expected = ['Homo sapiens', 'Mus musculus', 'Ursus arctos horribilis', 'Myrmecophaga tridactyla',
+                    'Panda']
         self.assertEqual(list(termset.view_set), expected)
         self.assertIsInstance(termset.view, SchemaView)