def test_equal(self): self.assertEqual(Expression("", "", ""), Expression("", "", "")) self.assertEqual( Expression(Child(This(), Fields("name")), "==", 4), Expression(Child(This(), Fields("name")), "==", 4), ) self.assertNotEqual(Expression("", "", ""), Expression(None, None, None))
class TestThis(TestCase): def setUp(self): self._this = This() self._data = {"data": [{"name": "jorge"}, {"name": "james"}]} def test_find(self): self.assertEqual(list(self._this.find(None)), [None]) self.assertEqual(list(self._this.find({})), [{}]) self.assertEqual(list(self._this.find(self._data)), [self._data]) self.assertEqual(list(self._this.find(34)), [34]) self.assertEqual(list(self._this.find("")), [""])
def test_node_not_equals(self): assert not Root() == This() assert not This() == Index(2) assert not Index(2) == Root() assert not Len() == Avg() assert not Sum() == Len() assert not Avg() == Sorted() assert not Sorted() == Sum() assert not And(Root(), Fields("name")) == And(Root(), Fields("other")) assert not Or(Root(), Fields("name")) == Or(Root(), Fields("other")) assert not Or(Root(), Fields("name")) == And(Root(), Fields("name"))
def test_find(self): data = {"value": 2, "other": 3} self.assertEqual( Expression(Child(This(), Fields("value")), "==", 2).find(data), True) self.assertEqual( Expression(Child(This(), Fields("other")), "<", 6).find(data), True) self.assertEqual( Expression(Child(This(), Fields("value")), None, 2).find(data), True) self.assertEqual( Expression(Child(This(), Fields("val")), ">=", 4).find({}), False)
def test_node_equals(self): assert Root() == Root() assert This() == This() assert Index(2) == Index(2) assert Len() == Len() assert Sum() == Sum() assert Avg() == Avg() assert Sorted() == Sorted() assert And(Root(), Fields("name")) == And(Root(), Fields("name")) assert Or(Root(), Fields("name")) == Or(Root(), Fields("name")) assert Child(Root(), Fields("name")) == Child(Root(), Fields("name")) assert Where(Root(), Fields("name")) == Where(Root(), Fields("name"))
def test_str_and_repr(self): self.assertEqual(str(Expression("", "", "")), "") self.assertEqual( str(Expression(Child(This(), Fields("name")), ">", 10)), "@.name>10") self.assertEqual(repr(Expression("", "", "")), "Expression(target='',op='',value='')") self.assertEqual( repr(Expression("name", ">", 10)), "Expression(target='name',op='>',value=10)", )
def test_repr_all_leaf(self): self.assertEqual(repr(Root()), "Root()") self.assertEqual(repr(This()), "This()") self.assertEqual(repr(Index(2)), "Index(2)") self.assertEqual(repr(AllIndex()), "AllIndex()") self.assertEqual(repr(Fields("name", "other")), "Fields('name','other')") self.assertEqual(repr(Slice(0, 3, 2)), "Slice(start=0,end=3,step=2)") self.assertEqual(repr(Len()), "Len()") self.assertEqual(repr(Sum()), "Sum()") self.assertEqual(repr(Avg()), "Avg()") self.assertEqual(repr(Sorted()), "Sorted()") self.assertEqual(repr(And(Fields("name"), Fields("id"))), "And(Fields('name'),Fields('id'))") self.assertEqual(repr(Or(Fields("name"), Fields("id"))), "Or(Fields('name'),Fields('id'))")
def test_str_all_leaf(self): self.assertEqual(str(Root()), "$") self.assertEqual(str(This()), "@") self.assertEqual(str(Index(1)), "[1]") self.assertEqual(str(Index("*")), "[*]") self.assertEqual(str(AllIndex()), "[*]") self.assertEqual(str(Fields("name", "other")), "name,other") self.assertEqual(str(Fields("name")), "name") self.assertEqual(str(Slice(0, 3, 2)), "[0:3:2]") self.assertEqual(str(Slice()), "[::]") self.assertEqual(str(Len()), "`len`") self.assertEqual(str(Sum()), "`sum`") self.assertEqual(str(Avg()), "`avg`") self.assertEqual(str(Sorted()), "`sorted`") self.assertEqual(str(And(Fields("name"), Fields("id"))), "name&id") self.assertEqual(str(Or(Fields("name"), Fields("id"))), "name|id")
def test_find(self): data = [{"value": 2}, {"other": 3}, {"value": 1}] exp = Expression(Child(This(), Fields("value")), "<=", 2) self.assertEqual(find(Filter(exp), data), [{"value": 2}, {"value": 1}])
def setUp(self): self._this = This() self._data = {"data": [{"name": "jorge"}, {"name": "james"}]}