1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
# Copyright (c) 2015 Mirantis, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
import yaql.tests
class TestBranching(yaql.tests.TestCase):
def test_switch(self):
expr = 'switch($ < 10 => 1, $ >= 10 and $ < 100 => 2, $ >= 100 => 3)'
self.assertEqual(3, self.eval(expr, data=123))
self.assertEqual(2, self.eval(expr, data=50))
self.assertEqual(1, self.eval(expr, data=-123))
def test_select_case(self):
expr = 'selectCase($ < 10, $ >= 10 and $ < 100)'
self.assertEqual(2, self.eval(expr, data=123))
self.assertEqual(1, self.eval(expr, data=50))
self.assertEqual(0, self.eval(expr, data=-123))
def test_select_all_cases(self):
expr = 'selectAllCases($ < 10, $ > 5)'
self.assertEqual([0], self.eval(expr, data=1))
self.assertEqual([0, 1], self.eval(expr, data=7))
self.assertEqual([1], self.eval(expr, data=12))
def test_examine(self):
expr = 'examine($ < 10, $ > 5)'
self.assertEqual([True, False], self.eval(expr, data=1))
self.assertEqual([True, True], self.eval(expr, data=7))
self.assertEqual([False, True], self.eval(expr, data=12))
def test_switch_case(self):
expr = "$.switchCase('a', 'b', 'c')"
self.assertEqual('a', self.eval(expr, data=0))
self.assertEqual('b', self.eval(expr, data=1))
self.assertEqual('c', self.eval(expr, data=3))
self.assertEqual('c', self.eval(expr, data=30))
self.assertEqual('c', self.eval(expr, data=-30))
def test_coalesce(self):
self.assertEqual(2, self.eval('coalesce($, 2)', data=None))
self.assertEqual(1, self.eval('coalesce($, 2)', data=1))
self.assertEqual(2, self.eval('coalesce($, $, 2)', data=None))
self.assertIsNone(self.eval('coalesce($)', data=None))
|
