add find operators

2018-06-13 02:29:32 +02:00
parent 38a59b921e
commit f9bb652034
3 changed files with 52 additions and 2 deletions
@@ -321,6 +321,22 @@ class TableTestCase(unittest.TestCase):
        ds = list(self.tbl.find(self.tbl.table.columns.temperature > 4))
        assert len(ds) == 3, ds

+    def test_find_dsl(self):
+        ds = list(self.tbl.find(place={'like': '%lw%'}))
+        assert len(ds) == 3, ds
+        ds = list(self.tbl.find(temperature={'>': 5}))
+        assert len(ds) == 2, ds
+        ds = list(self.tbl.find(temperature={'>=': 5}))
+        assert len(ds) == 3, ds
+        ds = list(self.tbl.find(temperature={'<': 0}))
+        assert len(ds) == 1, ds
+        ds = list(self.tbl.find(temperature={'<=': 0}))
+        assert len(ds) == 2, ds
+        ds = list(self.tbl.find(temperature={'!=': -1}))
+        assert len(ds) == 5, ds
+        ds = list(self.tbl.find(temperature={'between': [5, 8]}))
+        assert len(ds) == 3, ds
+
    def test_offset(self):
        ds = list(self.tbl.find(place=TEST_CITY_1, _offset=1))
        assert len(ds) == 2, ds