Added a couple of extra tests

Materials-Consortia · Mar 4, 2020 · 9d153a2 · 9d153a2
1 parent 2f77112
commit 9d153a2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/tests/filterparser/test_filterparser.py b/tests/filterparser/test_filterparser.py
@@ -187,6 +187,7 @@ def test_list_properties(self):
         # OPTIONAL:
         self.assertIsInstance(self.parse('elements HAS ONLY "H","He","Ga","Ta"'), Tree)
         self.assertIsInstance(self.parse('elements HAS ALL "H","He","Ga","Ta"'), Tree)
+        self.assertIsInstance(self.parse('elements HAS ANY "H","He","Ga","Ta"'), Tree)
         self.assertIsInstance(
             self.parse(
                 'elements:_exmpl_element_counts HAS "H":6 AND '

diff --git a/tests/filtertransformers/test_mongo.py b/tests/filtertransformers/test_mongo.py
@@ -293,6 +293,11 @@ def test_list_properties(self):
             {"elements": {"$in": ["H", "He", "Ga", "Ta"]}},
         )
 
+        self.assertEqual(
+            self.transform('elements HAS ALL "H","He","Ga","Ta"'),
+            {"elements": {"$all": ["H", "He", "Ga", "Ta"]}},
+        )
+
         self.assertEqual(
             self.transform(
                 'elements HAS "H" AND elements HAS ALL "H","He","Ga","Ta" AND elements HAS '