Damien Martin-Guillerez | f88f4d8 | 2015-09-25 13:56:55 +0000 | [diff] [blame] | 1 | // Copyright 2014 The Bazel Authors. All rights reserved. |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 2 | // |
| 3 | // Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | // you may not use this file except in compliance with the License. |
| 5 | // You may obtain a copy of the License at |
| 6 | // |
| 7 | // http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | // |
| 9 | // Unless required by applicable law or agreed to in writing, software |
| 10 | // distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | // See the License for the specific language governing permissions and |
| 13 | // limitations under the License. |
| 14 | package com.google.devtools.build.lib.syntax; |
| 15 | |
| 16 | import static com.google.common.truth.Truth.assertThat; |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 17 | import static org.junit.Assert.assertEquals; |
| 18 | import static org.junit.Assert.assertNotNull; |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 19 | |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 20 | import com.google.common.collect.ImmutableMap; |
Han-Wen Nienhuys | ceae8c5 | 2015-09-22 16:24:45 +0000 | [diff] [blame] | 21 | import com.google.devtools.build.lib.syntax.util.EvaluationTestCase; |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 22 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 23 | import org.junit.Test; |
| 24 | import org.junit.runner.RunWith; |
| 25 | import org.junit.runners.JUnit4; |
| 26 | |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 27 | import java.util.ArrayList; |
| 28 | import java.util.List; |
| 29 | import java.util.Map; |
| 30 | |
| 31 | /** |
| 32 | * A test class for functions and scoping. |
| 33 | */ |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 34 | @RunWith(JUnit4.class) |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 35 | public class FunctionTest extends EvaluationTestCase { |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 36 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 37 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 38 | public void testFunctionDef() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 39 | eval("def func(a,b,c):", |
| 40 | " a = 1", |
| 41 | " b = a\n"); |
| 42 | UserDefinedFunction stmt = (UserDefinedFunction) lookup("func"); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 43 | assertNotNull(stmt); |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 44 | assertThat(stmt.getName()).isEqualTo("func"); |
| 45 | assertThat(stmt.getFunctionSignature().getSignature().getShape().getMandatoryPositionals()) |
| 46 | .isEqualTo(3); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 47 | assertThat(stmt.getStatements()).hasSize(2); |
| 48 | } |
| 49 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 50 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 51 | public void testFunctionDefDuplicateArguments() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 52 | setFailFast(false); |
| 53 | parseFile("def func(a,b,a):", |
| 54 | " a = 1\n"); |
Ulf Adams | c708f96 | 2015-10-22 12:02:28 +0000 | [diff] [blame] | 55 | assertContainsError("duplicate parameter name in function definition"); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 56 | } |
| 57 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 58 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 59 | public void testFunctionDefCallOuterFunc() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 60 | List<Object> params = new ArrayList<>(); |
| 61 | createOuterFunction(params); |
| 62 | eval("def func(a):", |
| 63 | " outer_func(a)", |
| 64 | "func(1)", |
| 65 | "func(2)"); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 66 | assertThat(params).containsExactly(1, 2).inOrder(); |
| 67 | } |
| 68 | |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 69 | private void createOuterFunction(final List<Object> params) throws Exception { |
Francois-Rene Rideau | 4feb160 | 2015-03-18 19:49:13 +0000 | [diff] [blame] | 70 | BaseFunction outerFunc = new BaseFunction("outer_func") { |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 71 | @Override |
| 72 | public Object call(List<Object> args, Map<String, Object> kwargs, FuncallExpression ast, |
| 73 | Environment env) throws EvalException, InterruptedException { |
| 74 | params.addAll(args); |
Francois-Rene Rideau | 0f7ba34 | 2015-08-31 16:16:21 +0000 | [diff] [blame] | 75 | return Runtime.NONE; |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 76 | } |
| 77 | }; |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 78 | update("outer_func", outerFunc); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 79 | } |
| 80 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 81 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 82 | public void testFunctionDefNoEffectOutsideScope() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 83 | update("a", 1); |
| 84 | eval("def func():", |
| 85 | " a = 2", |
| 86 | "func()\n"); |
| 87 | assertEquals(1, lookup("a")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 88 | } |
| 89 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 90 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 91 | public void testFunctionDefGlobalVaribleReadInFunction() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 92 | eval("a = 1", |
| 93 | "def func():", |
| 94 | " b = a", |
| 95 | " return b", |
| 96 | "c = func()\n"); |
| 97 | assertEquals(1, lookup("c")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 98 | } |
| 99 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 100 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 101 | public void testFunctionDefLocalGlobalScope() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 102 | eval("a = 1", |
| 103 | "def func():", |
| 104 | " a = 2", |
| 105 | " b = a", |
| 106 | " return b", |
| 107 | "c = func()\n"); |
| 108 | assertEquals(2, lookup("c")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 109 | } |
| 110 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 111 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 112 | public void testFunctionDefLocalVariableReferencedBeforeAssignment() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 113 | checkEvalErrorContains("Variable 'a' is referenced before assignment.", |
| 114 | "a = 1", |
| 115 | "def func():", |
| 116 | " b = a", |
| 117 | " a = 2", |
| 118 | " return b", |
| 119 | "c = func()\n"); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 120 | } |
| 121 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 122 | @Test |
Jon Brandvein | 83e3acb | 2016-08-11 18:53:26 +0000 | [diff] [blame] | 123 | public void testFunctionDefLocalVariableReferencedInCallBeforeAssignment() throws Exception { |
| 124 | checkEvalErrorContains("Variable 'a' is referenced before assignment.", |
| 125 | "def dummy(x):", |
| 126 | " pass", |
| 127 | "a = 1", |
| 128 | "def func():", |
| 129 | " dummy(a)", |
| 130 | " a = 2", |
| 131 | "func()\n"); |
| 132 | } |
| 133 | |
| 134 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 135 | public void testFunctionDefLocalVariableReferencedAfterAssignment() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 136 | eval("a = 1", |
| 137 | "def func():", |
| 138 | " a = 2", |
| 139 | " b = a", |
| 140 | " a = 3", |
| 141 | " return b", |
| 142 | "c = func()\n"); |
| 143 | assertEquals(2, lookup("c")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 144 | } |
| 145 | |
| 146 | @SuppressWarnings("unchecked") |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 147 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 148 | public void testSkylarkGlobalComprehensionIsAllowed() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 149 | eval("a = [i for i in [1, 2, 3]]\n"); |
| 150 | assertThat((Iterable<Object>) lookup("a")).containsExactly(1, 2, 3).inOrder(); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 151 | } |
| 152 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 153 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 154 | public void testFunctionReturn() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 155 | eval("def func():", |
| 156 | " return 2", |
| 157 | "b = func()\n"); |
| 158 | assertEquals(2, lookup("b")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 159 | } |
| 160 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 161 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 162 | public void testFunctionReturnFromALoop() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 163 | eval("def func():", |
| 164 | " for i in [1, 2, 3, 4, 5]:", |
| 165 | " return i", |
| 166 | "b = func()\n"); |
| 167 | assertEquals(1, lookup("b")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 168 | } |
| 169 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 170 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 171 | public void testFunctionExecutesProperly() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 172 | eval("def func(a):", |
| 173 | " b = 1", |
| 174 | " if a:", |
| 175 | " b = 2", |
| 176 | " return b", |
| 177 | "c = func(0)", |
| 178 | "d = func(1)\n"); |
| 179 | assertEquals(1, lookup("c")); |
| 180 | assertEquals(2, lookup("d")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 181 | } |
| 182 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 183 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 184 | public void testFunctionCallFromFunction() throws Exception { |
| 185 | final List<Object> params = new ArrayList<>(); |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 186 | createOuterFunction(params); |
| 187 | eval("def func2(a):", |
| 188 | " outer_func(a)", |
| 189 | "def func1(b):", |
| 190 | " func2(b)", |
| 191 | "func1(1)", |
| 192 | "func1(2)\n"); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 193 | assertThat(params).containsExactly(1, 2).inOrder(); |
| 194 | } |
| 195 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 196 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 197 | public void testFunctionCallFromFunctionReadGlobalVar() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 198 | eval("a = 1", |
| 199 | "def func2():", |
| 200 | " return a", |
| 201 | "def func1():", |
| 202 | " return func2()", |
| 203 | "b = func1()\n"); |
| 204 | assertEquals(1, lookup("b")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 205 | } |
| 206 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 207 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 208 | public void testSingleLineFunction() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 209 | eval("def func(): return 'a'", |
| 210 | "s = func()\n"); |
| 211 | assertEquals("a", lookup("s")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 212 | } |
| 213 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 214 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 215 | public void testFunctionReturnsDictionary() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 216 | eval("def func(): return {'a' : 1}", |
| 217 | "d = func()", |
| 218 | "a = d['a']\n"); |
| 219 | assertEquals(1, lookup("a")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 220 | } |
| 221 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 222 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 223 | public void testFunctionReturnsList() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 224 | eval("def func(): return [1, 2, 3]", |
| 225 | "d = func()", |
| 226 | "a = d[1]\n"); |
| 227 | assertEquals(2, lookup("a")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 228 | } |
| 229 | |
| 230 | @SuppressWarnings("unchecked") |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 231 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 232 | public void testFunctionListArgumentsAreImmutable() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 233 | eval("l = [1]", |
| 234 | "def func(l):", |
| 235 | " l += [2]", |
| 236 | "func(l)"); |
| 237 | assertThat((Iterable<Object>) lookup("l")).containsExactly(1); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 238 | } |
| 239 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 240 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 241 | public void testFunctionDictArgumentsAreImmutable() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 242 | eval("d = {'a' : 1}", |
| 243 | "def func(d):", |
| 244 | " d += {'a' : 2}", |
| 245 | "func(d)"); |
| 246 | assertEquals(ImmutableMap.of("a", 1), lookup("d")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 247 | } |
| 248 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 249 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 250 | public void testFunctionNameAliasing() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 251 | eval("def func(a):", |
| 252 | " return a + 1", |
| 253 | "alias = func", |
| 254 | "r = alias(1)"); |
| 255 | assertEquals(2, lookup("r")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 256 | } |
| 257 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 258 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 259 | public void testCallingFunctionsWithMixedModeArgs() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 260 | eval("def func(a, b, c):", |
| 261 | " return a + b + c", |
| 262 | "v = func(1, c = 2, b = 3)"); |
| 263 | assertEquals(6, lookup("v")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 264 | } |
| 265 | |
| 266 | private String functionWithOptionalArgs() { |
| 267 | return "def func(a, b = None, c = None):\n" |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 268 | + " r = a + 'a'\n" |
| 269 | + " if b:\n" |
| 270 | + " r += 'b'\n" |
| 271 | + " if c:\n" |
| 272 | + " r += 'c'\n" |
| 273 | + " return r\n"; |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 274 | } |
| 275 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 276 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 277 | public void testWhichOptionalArgsAreDefinedForFunctions() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 278 | eval(functionWithOptionalArgs(), |
| 279 | "v1 = func('1', 1, 1)", |
| 280 | "v2 = func(b = 2, a = '2', c = 2)", |
| 281 | "v3 = func('3')", |
| 282 | "v4 = func('4', c = 1)\n"); |
| 283 | assertEquals("1abc", lookup("v1")); |
| 284 | assertEquals("2abc", lookup("v2")); |
| 285 | assertEquals("3a", lookup("v3")); |
| 286 | assertEquals("4ac", lookup("v4")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 287 | } |
| 288 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 289 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 290 | public void testDefaultArguments() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 291 | eval("def func(a, b = 'b', c = 'c'):", |
| 292 | " return a + b + c", |
| 293 | "v1 = func('a', 'x', 'y')", |
| 294 | "v2 = func(b = 'x', a = 'a', c = 'y')", |
| 295 | "v3 = func('a')", |
| 296 | "v4 = func('a', c = 'y')\n"); |
| 297 | assertEquals("axy", lookup("v1")); |
| 298 | assertEquals("axy", lookup("v2")); |
| 299 | assertEquals("abc", lookup("v3")); |
| 300 | assertEquals("aby", lookup("v4")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 301 | } |
| 302 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 303 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 304 | public void testDefaultArgumentsInsufficientArgNum() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 305 | checkEvalError("insufficient arguments received by func(a, b = \"b\", c = \"c\") " |
Francois-Rene Rideau | 4feb160 | 2015-03-18 19:49:13 +0000 | [diff] [blame] | 306 | + "(got 0, expected at least 1)", |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 307 | "def func(a, b = 'b', c = 'c'):", |
| 308 | " return a + b + c", |
| 309 | "func()"); |
| 310 | } |
| 311 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 312 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 313 | public void testKwargs() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 314 | eval("def foo(a, b = 'b', *, c, d = 'd'):", |
| 315 | " return a + b + c + d", |
| 316 | "args = {'a': 'x', 'c': 'z'}", |
| 317 | "v1 = foo(**args)", |
| 318 | "v2 = foo('x', c = 'c', d = 'e', **{'b': 'y'})", |
| 319 | "v3 = foo(c = 'z', a = 'x', **{'b': 'y', 'd': 'f'})"); |
| 320 | assertEquals("xbzd", lookup("v1")); |
| 321 | assertEquals("xyce", lookup("v2")); |
| 322 | assertEquals("xyzf", lookup("v3")); |
| 323 | UserDefinedFunction foo = (UserDefinedFunction) lookup("foo"); |
Francois-Rene Rideau | 012f789 | 2015-03-31 17:27:01 +0000 | [diff] [blame] | 324 | assertEquals("foo(a, b = \"b\", *, c, d = \"d\")", foo.toString()); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 325 | } |
| 326 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 327 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 328 | public void testKwargsBadKey() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 329 | checkEvalError("Keywords must be strings, not int", |
| 330 | "def func(a, b): return a + b", |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 331 | "func('a', **{3: 1})"); |
| 332 | } |
| 333 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 334 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 335 | public void testKwargsIsNotDict() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 336 | checkEvalError("Argument after ** must be a dictionary, not int", |
| 337 | "def func(a, b): return a + b", |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 338 | "func('a', **42)"); |
| 339 | } |
| 340 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 341 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 342 | public void testKwargsCollision() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 343 | checkEvalError("argument 'b' passed both by position and by name in call to func(a, b)", |
| 344 | "def func(a, b): return a + b", |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 345 | "func('a', 'b', **{'b': 'foo'})"); |
| 346 | } |
| 347 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 348 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 349 | public void testKwargsCollisionWithNamed() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 350 | checkEvalError("duplicate keyword 'b' in call to func", |
| 351 | "def func(a, b): return a + b", |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 352 | "func('a', b = 'b', **{'b': 'foo'})"); |
| 353 | } |
| 354 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 355 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 356 | public void testDefaultArguments2() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 357 | eval("a = 2", |
| 358 | "def foo(x=a): return x", |
| 359 | "def bar():", |
| 360 | " a = 3", |
| 361 | " return foo()", |
| 362 | "v = bar()\n"); |
| 363 | assertEquals(2, lookup("v")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 364 | } |
| 365 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 366 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 367 | public void testMixingPositionalOptional() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 368 | eval("def f(name, value = '', optional = ''): return value", |
| 369 | "v = f('name', 'value')\n"); |
| 370 | assertEquals("value", lookup("v")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 371 | } |
| 372 | |
Han-Wen Nienhuys | ccf19ea | 2015-02-27 15:53:24 +0000 | [diff] [blame] | 373 | @Test |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 374 | public void testStarArg() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 375 | eval("def f(name, value = '1', optional = '2'): return name + value + optional", |
| 376 | "v1 = f(*['name', 'value'])", |
| 377 | "v2 = f('0', *['name', 'value'])", |
| 378 | "v3 = f('0', *['b'], optional = '3')", |
| 379 | "v4 = f(*[],name='a')\n"); |
| 380 | assertEquals("namevalue2", lookup("v1")); |
| 381 | assertEquals("0namevalue", lookup("v2")); |
| 382 | assertEquals("0b3", lookup("v3")); |
| 383 | assertEquals("a12", lookup("v4")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 384 | } |
| 385 | |
Francois-Rene Rideau | 4feb160 | 2015-03-18 19:49:13 +0000 | [diff] [blame] | 386 | @Test |
| 387 | public void testStarParam() throws Exception { |
Francois-Rene Rideau | 5f3e30c | 2015-04-10 19:08:39 +0000 | [diff] [blame] | 388 | eval("def f(name, value = '1', *rest, mandatory, optional = '2'):", |
| 389 | " r = name + value + mandatory + optional + '|'", |
| 390 | " for x in rest: r += x", |
| 391 | " return r", |
| 392 | "v1 = f('a', 'b', mandatory = 'z')", |
| 393 | "v2 = f('a', 'b', 'c', 'd', mandatory = 'z')", |
| 394 | "v3 = f('a', *['b', 'c', 'd'], mandatory = 'y', optional = 'z')", |
| 395 | "v4 = f(*['a'], **{'value': 'b', 'mandatory': 'c'})", |
| 396 | "v5 = f('a', 'b', 'c', *['d', 'e'], mandatory = 'f', **{'optional': 'g'})\n"); |
| 397 | assertEquals("abz2|", lookup("v1")); |
| 398 | assertEquals("abz2|cd", lookup("v2")); |
| 399 | assertEquals("abyz|cd", lookup("v3")); |
| 400 | assertEquals("abc2|", lookup("v4")); |
| 401 | assertEquals("abfg|cde", lookup("v5")); |
Ulf Adams | 89f012d | 2015-02-26 13:39:28 +0000 | [diff] [blame] | 402 | } |
| 403 | } |