// Copyright 2006 The Bazel Authors. All Rights Reserved. // // 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. package com.google.devtools.build.lib.syntax; import static com.google.common.truth.Truth.assertThat; import com.google.common.collect.ImmutableMap; import com.google.common.collect.Iterables; import com.google.devtools.build.lib.syntax.SkylarkList.MutableList; import com.google.devtools.build.lib.syntax.util.EvaluationTestCase; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.junit.runners.JUnit4; /** * Tests for MethodLibrary. */ @RunWith(JUnit4.class) public class MethodLibraryTest extends EvaluationTestCase { private static final String LINE_SEPARATOR = System.lineSeparator(); @Before public final void setFailFast() throws Exception { setFailFast(true); } @Test public void testMinWithInvalidArgs() throws Exception { new SkylarkTest() .testIfExactError("type 'int' is not iterable", "min(1)") .testIfExactError("expected at least one item", "min([])"); } @Test public void testMinWithString() throws Exception { new SkylarkTest() .testStatement("min('abcdefxyz')", "a") .testStatement("min('test', 'xyz')", "test"); } @Test public void testMinWithList() throws Exception { new BothModesTest() .testEval("min([4, 5], [1])", "[1]") .testEval("min([1, 2], [3])", "[1, 2]") .testEval("min([1, 5], [1, 6], [2, 4], [0, 6])", "[0, 6]") .testStatement("min([-1])", -1) .testStatement("min([5, 2, 3])", 2); } @Test public void testMinWithDict() throws Exception { new BothModesTest().testStatement("min({1: 2, -1 : 3})", -1).testStatement("min({2: None})", 2); } @Test public void testMinWithSet() throws Exception { new BothModesTest() .testStatement("min(depset([-1]))", -1) .testStatement("min(depset([5, 2, 3]))", 2); } @Test public void testMinWithPositionalArguments() throws Exception { new BothModesTest().testStatement("min(-1, 2)", -1).testStatement("min(5, 2, 3)", 2); } @Test public void testMinWithSameValues() throws Exception { new BothModesTest() .testStatement("min(1, 1, 1, 1, 1, 1)", 1) .testStatement("min([1, 1, 1, 1, 1, 1])", 1); } @Test public void testMinWithDifferentTypes() throws Exception { new BothModesTest() .testIfExactError("Cannot compare int with string", "min(1, '2', True)") .testIfExactError("Cannot compare int with string", "min([1, '2', True])"); } @Test public void testMaxWithInvalidArgs() throws Exception { new BothModesTest() .testIfExactError("type 'int' is not iterable", "max(1)") .testIfExactError("expected at least one item", "max([])"); } @Test public void testMaxWithString() throws Exception { new BothModesTest() .testStatement("max('abcdefxyz')", "z") .testStatement("max('test', 'xyz')", "xyz"); } @Test public void testMaxWithList() throws Exception { new BothModesTest() .testEval("max([1, 2], [5])", "[5]") .testStatement("max([-1])", -1) .testStatement("max([5, 2, 3])", 5); } @Test public void testMaxWithDict() throws Exception { new BothModesTest().testStatement("max({1: 2, -1 : 3})", 1).testStatement("max({2: None})", 2); } @Test public void testMaxWithSet() throws Exception { new BothModesTest() .testStatement("max(depset([-1]))", -1) .testStatement("max(depset([5, 2, 3]))", 5); } @Test public void testMaxWithPositionalArguments() throws Exception { new BothModesTest().testStatement("max(-1, 2)", 2).testStatement("max(5, 2, 3)", 5); } @Test public void testMaxWithSameValues() throws Exception { new BothModesTest() .testStatement("max(1, 1, 1, 1, 1, 1)", 1) .testStatement("max([1, 1, 1, 1, 1, 1])", 1); } @Test public void testMaxWithDifferentTypes() throws Exception { new BothModesTest() .testIfExactError("Cannot compare int with string", "max(1, '2', True)") .testIfExactError("Cannot compare int with string", "max([1, '2', True])"); } @Test public void testSplitLines_EmptyLine() throws Exception { new BothModesTest().testEval("''.splitlines()", "[]").testEval("'\\n'.splitlines()", "['']"); } @Test public void testSplitLines_StartsWithLineBreak() throws Exception { new BothModesTest().testEval("'\\ntest'.splitlines()", "['', 'test']"); } @Test public void testSplitLines_EndsWithLineBreak() throws Exception { new BothModesTest().testEval("'test\\n'.splitlines()", "['test']"); } @Test public void testSplitLines_DifferentLineBreaks() throws Exception { new BothModesTest() .testEval("'this\\nis\\na\\ntest'.splitlines()", "['this', 'is', 'a', 'test']"); } @Test public void testSplitLines_OnlyLineBreaks() throws Exception { new BothModesTest() .testEval("'\\n\\n\\n'.splitlines()", "['', '', '']") .testEval("'\\r\\r\\r'.splitlines()", "['', '', '']") .testEval("'\\n\\r\\n\\r'.splitlines()", "['', '', '']") .testEval("'\\r\\n\\r\\n\\r\\n'.splitlines()", "['', '', '']"); } @Test public void testSplitLines_EscapedSequences() throws Exception { new BothModesTest().testEval("'\\n\\\\n\\\\\\n'.splitlines()", "['', '\\\\n\\\\']"); } @Test public void testSplitLines_KeepEnds() throws Exception { new BothModesTest() .testEval("''.splitlines(True)", "[]") .testEval("'\\n'.splitlines(True)", "['\\n']") .testEval( "'this\\nis\\r\\na\\rtest'.splitlines(True)", "['this\\n', 'is\\r\\n', 'a\\r', 'test']") .testEval("'\\ntest'.splitlines(True)", "['\\n', 'test']") .testEval("'test\\n'.splitlines(True)", "['test\\n']") .testEval("'\\n\\\\n\\\\\\n'.splitlines(True)", "['\\n', '\\\\n\\\\\\n']"); } @Test public void testStringIsAlnum() throws Exception { new BothModesTest() .testStatement("''.isalnum()", false) .testStatement("'a0 33'.isalnum()", false) .testStatement("'1'.isalnum()", true) .testStatement("'a033'.isalnum()", true); } @Test public void testStringIsDigit() throws Exception { new BothModesTest() .testStatement("''.isdigit()", false) .testStatement("' '.isdigit()", false) .testStatement("'a'.isdigit()", false) .testStatement("'0234325.33'.isdigit()", false) .testStatement("'1'.isdigit()", true) .testStatement("'033'.isdigit()", true); } @Test public void testStringIsSpace() throws Exception { new BothModesTest() .testStatement("''.isspace()", false) .testStatement("'a'.isspace()", false) .testStatement("'1'.isspace()", false) .testStatement("'\\ta\\n'.isspace()", false) .testStatement("' '.isspace()", true) .testStatement("'\\t\\n'.isspace()", true); } @Test public void testStringIsLower() throws Exception { new BothModesTest() .testStatement("''.islower()", false) .testStatement("' '.islower()", false) .testStatement("'1'.islower()", false) .testStatement("'Almost'.islower()", false) .testStatement("'abc'.islower()", true) .testStatement("' \\nabc'.islower()", true) .testStatement("'abc def\\n'.islower()", true) .testStatement("'\\ta\\n'.islower()", true); } @Test public void testStringIsUpper() throws Exception { new BothModesTest() .testStatement("''.isupper()", false) .testStatement("' '.isupper()", false) .testStatement("'1'.isupper()", false) .testStatement("'aLMOST'.isupper()", false) .testStatement("'ABC'.isupper()", true) .testStatement("' \\nABC'.isupper()", true) .testStatement("'ABC DEF\\n'.isupper()", true) .testStatement("'\\tA\\n'.isupper()", true); } @Test public void testStringIsTitle() throws Exception { new BothModesTest() .testStatement("''.istitle()", false) .testStatement("' '.istitle()", false) .testStatement("'134'.istitle()", false) .testStatement("'almost Correct'.istitle()", false) .testStatement("'1nope Nope Nope'.istitle()", false) .testStatement("'NO Way'.istitle()", false) .testStatement("'T'.istitle()", true) .testStatement("'Correct'.istitle()", true) .testStatement("'Very Correct! Yes\\nIndeed1X'.istitle()", true) .testStatement("'1234Ab Ab'.istitle()", true) .testStatement("'\\tA\\n'.istitle()", true); } @Test public void testAllWithEmptyValue() throws Exception { new BothModesTest() .testStatement("all('')", true) .testStatement("all([])", true) .testIfExactError("type 'NoneType' is not iterable", "any(None)"); } @Test public void testAllWithPrimitiveType() throws Exception { new BothModesTest().testStatement("all('test')", true).testIfErrorContains("", "all(1)"); } @Test public void testAllWithList() throws Exception { new BothModesTest() .testStatement("all([False])", false) .testStatement("all([True, False])", false) .testStatement("all([False, False])", false) .testStatement("all([False, True])", false) .testStatement("all(['', True])", false) .testStatement("all([0, True])", false) .testStatement("all([[], True])", false) .testStatement("all([True, 't', 1])", true); } @Test public void testAllWithSet() throws Exception { new BothModesTest() .testStatement("all(depset([0]))", false) .testStatement("all(depset([1, 0]))", false) .testStatement("all(depset([1]))", true); } @Test public void testAllWithDict() throws Exception { new BothModesTest() .testStatement("all({1 : None})", true) .testStatement("all({None : 1})", false); } @Test public void testAnyWithEmptyValue() throws Exception { new BothModesTest() .testStatement("any('')", false) .testStatement("any([])", false) .testIfExactError("type 'NoneType' is not iterable", "any(None)"); } @Test public void testAnyWithPrimitiveType() throws Exception { new BothModesTest().testStatement("any('test')", true).testIfErrorContains("", "any(1)"); } @Test public void testAnyWithList() throws Exception { new BothModesTest() .testStatement("any([False])", false) .testStatement("any([0])", false) .testStatement("any([''])", false) .testStatement("any([[]])", false) .testStatement("any([True, False])", true) .testStatement("any([False, False])", false) .testStatement("any([False, '', 0])", false) .testStatement("any([False, '', 42])", true); } @Test public void testAnyWithSet() throws Exception { new BothModesTest() .testStatement("any(depset([0]))", false) .testStatement("any(depset([1, 0]))", true); } @Test public void testAnyWithDict() throws Exception { new BothModesTest() .testStatement("any({1 : None, '' : None})", true) .testStatement("any({None : 1, '' : 2})", false); } @Test public void testStackTraceLocation() throws Exception { new SkylarkTest() .testIfErrorContains( "Traceback (most recent call last):" + LINE_SEPARATOR + "\tFile \"\", line 8" + LINE_SEPARATOR + "\t\tfoo()" + LINE_SEPARATOR + "\tFile \"\", line 2, in foo" + LINE_SEPARATOR + "\t\tbar(1)" + LINE_SEPARATOR + "\tFile \"\", line 7, in bar" + LINE_SEPARATOR + "\t\t\"test\".index(x)", "def foo():", " bar(1)", "def bar(x):", " if x == 1:", " a = x", " b = 2", " 'test'.index(x)", "foo()"); } @Test public void testStackTraceWithIf() throws Exception { new SkylarkTest() .testIfErrorContains( "File \"\", line 5" + LINE_SEPARATOR + "\t\tfoo()" + LINE_SEPARATOR + "\tFile \"\", line 3, in foo" + LINE_SEPARATOR + "\t\ts[0]", "def foo():", " s = depset()", " if s[0] == 1:", " x = 1", "foo()"); } @Test public void testStackTraceSkipBuiltInOnly() throws Exception { // The error message should not include the stack trace when there is // only one built-in function. new BothModesTest() .testIfExactError( "method string.index(sub: string, start: int, end: int or NoneType) is not applicable " + "for arguments (int, int, NoneType): 'sub' is 'int', but should be 'string'", "'test'.index(1)"); } @Test public void testStackTrace() throws Exception { // Unlike SkylarintegrationTests#testStackTraceErrorInFunction(), this test // has neither a BUILD nor a bzl file. new SkylarkTest() .testIfExactError( "Traceback (most recent call last):" + LINE_SEPARATOR + "\tFile \"\", line 6" + LINE_SEPARATOR + "\t\tfoo()" + LINE_SEPARATOR + "\tFile \"\", line 2, in foo" + LINE_SEPARATOR + "\t\tbar(1)" + LINE_SEPARATOR + "\tFile \"\", line 5, in bar" + LINE_SEPARATOR + "\t\t\"test\".index(x)" + LINE_SEPARATOR + "method string.index(sub: string, start: int, end: int or NoneType) " + "is not applicable " + "for arguments (int, int, NoneType): 'sub' is 'int', but should be 'string'", "def foo():", " bar(1)", "def bar(x):", " if 1 == 1:", " 'test'.index(x)", "foo()"); } @Test public void testBuiltinFunctionErrorMessage() throws Exception { new BothModesTest() .testIfErrorContains( "substring \"z\" not found in \"abc\"", "'abc'.index('z')") .testIfErrorContains( "method string.startswith(sub: string, start: int, end: int or NoneType) is not " + "applicable for arguments (int, int, NoneType): 'sub' is 'int', " + "but should be 'string'", "'test'.startswith(1)") .testIfErrorContains( "expected value of type 'list(object)' for parameter args in dict(), " + "but got \"a\" (string)", "dict('a')"); } @Test public void testHasAttr() throws Exception { new SkylarkTest() .testStatement("hasattr(depset(), 'union')", Boolean.TRUE) .testStatement("hasattr('test', 'count')", Boolean.TRUE) .testStatement("hasattr(dict(a = 1, b = 2), 'items')", Boolean.TRUE) .testStatement("hasattr({}, 'items')", Boolean.TRUE); } @Test public void testGetAttrMissingField() throws Exception { new SkylarkTest() .testIfExactError( "object of type 'string' has no attribute \"not_there\"", "getattr('a string', 'not_there')") .testStatement("getattr('a string', 'not_there', 'use this')", "use this") .testStatement("getattr('a string', 'not there', None)", Runtime.NONE); } @Test public void testGetAttrWithMethods() throws Exception { String msg = "object of type 'string' has no attribute \"count\", however, " + "a method of that name exists"; new SkylarkTest() .testIfExactError(msg, "getattr('a string', 'count')") .testStatement("getattr('a string', 'count', 'default')", "default"); } @Test public void testDir() throws Exception { new SkylarkTest() .testStatement( "str(dir({}))", "[\"clear\", \"get\", \"items\", \"keys\"," + " \"pop\", \"popitem\", \"setdefault\", \"update\", \"values\"]"); } @Test public void testBoolean() throws Exception { new BothModesTest().testStatement("False", Boolean.FALSE).testStatement("True", Boolean.TRUE); } @Test public void testBooleanUnsupportedOperationFails() throws Exception { new BothModesTest() .testIfErrorContains("unsupported operand type(s) for +: 'bool' and 'bool'", "True + True"); } @Test public void testPyStringJoin() throws Exception { new BothModesTest().testStatement("'-'.join(['a', 'b', 'c'])", "a-b-c"); } @Test public void testPyStringGlobalJoin() throws Exception { new BothModesTest() .testIfErrorContains("name 'join' is not defined", "join(' ', ['a', 'b', 'c'])") .testStatement("' '.join(['a', 'b', 'c'])", "a b c"); } @Test public void testPyStringJoinCompr() throws Exception { new BothModesTest() .testStatement("''.join([(x + '*') for x in ['a', 'b', 'c']])", "a*b*c*") .testStatement( "''.join([(y + '*' + z + '|') " + "for y in ['a', 'b', 'c'] for z in ['d', 'e']])", "a*d|a*e|b*d|b*e|c*d|c*e|"); } @Test public void testPyStringLower() throws Exception { new BothModesTest().testStatement("'Blah Blah'.lower()", "blah blah"); } @Test public void testPyStringUpper() throws Exception { new BothModesTest() .testStatement("'ein bier'.upper()", "EIN BIER") .testStatement("''.upper()", ""); } @Test public void testPyStringReplace() throws Exception { new BothModesTest() .testStatement("'banana'.replace('a', 'e')", "benene") .testStatement("'banana'.replace('a', '$()')", "b$()n$()n$()") .testStatement("'banana'.replace('a', '$')", "b$n$n$") .testStatement("'banana'.replace('a', '\\\\')", "b\\n\\n\\") .testStatement("'b$()n$()n$()'.replace('$()', '$($())')", "b$($())n$($())n$($())") .testStatement("'b\\\\n\\\\n\\\\'.replace('\\\\', '$()')", "b$()n$()n$()"); } @Test public void testPyStringReplace2() throws Exception { new BothModesTest().testStatement("'banana'.replace('a', 'e', 2)", "benena"); } @Test public void testPyStringSplit() throws Exception { new BothModesTest().testEval("'h i'.split(' ')", "['h', 'i']"); } @Test public void testPyStringSplit2() throws Exception { new BothModesTest().testEval("'h i p'.split(' ')", "['h', 'i', 'p']"); } @Test public void testPyStringSplit3() throws Exception { new BothModesTest().testEval("'a,e,i,o,u'.split(',', 2)", "['a', 'e', 'i,o,u']"); } @Test public void testPyStringSplitNoSep() throws Exception { new BothModesTest() .testEval("' 1 2 3 '.split(' ')", "['', '', '1', '', '2', '', '3', '', '']"); } @Test public void testPyStringRSplitRegex() throws Exception { new BothModesTest() .testEval("'foo/bar.lisp'.rsplit('.')", "['foo/bar', 'lisp']") .testEval("'foo/bar.?lisp'.rsplit('.?')", "['foo/bar', 'lisp']") .testEval("'fwe$foo'.rsplit('$')", "['fwe', 'foo']") .testEval("'windows'.rsplit('\\w')", "['windows']"); } @Test public void testPyStringRSplitNoMatch() throws Exception { new BothModesTest() .testEval("''.rsplit('o')", "['']") .testEval("'google'.rsplit('x')", "['google']"); } @Test public void testPyStringRSplitSeparator() throws Exception { new BothModesTest() .testEval("'xxxxxx'.rsplit('x')", "['', '', '', '', '', '', '']") .testEval("'xxxxxx'.rsplit('x', 1)", "['xxxxx', '']") .testEval("'xxxxxx'.rsplit('x', 2)", "['xxxx', '', '']") .testEval("'xxxxxx'.rsplit('x', 3)", "['xxx', '', '', '']") .testEval("'xxxxxx'.rsplit('x', 4)", "['xx', '', '', '', '']") .testEval("'xxxxxx'.rsplit('x', 5)", "['x', '', '', '', '', '']") .testEval("'xxxxxx'.rsplit('x', 6)", "['', '', '', '', '', '', '']") .testEval("'xxxxxx'.rsplit('x', 7)", "['', '', '', '', '', '', '']"); } @Test public void testPyStringRSplitLongerSep() throws Exception { new BothModesTest() .testEval("'abcdabef'.rsplit('ab')", "['', 'cd', 'ef']") .testEval("'google_or_gogol'.rsplit('go')", "['', 'ogle_or_', '', 'l']"); } @Test public void testPyStringRSplitMaxSplit() throws Exception { new BothModesTest() .testEval("'google'.rsplit('o')", "['g', '', 'gle']") .testEval("'google'.rsplit('o')", "['g', '', 'gle']") .testEval("'google'.rsplit('o', 1)", "['go', 'gle']") .testEval("'google'.rsplit('o', 2)", "['g', '', 'gle']") .testEval("'google'.rsplit('o', 3)", "['g', '', 'gle']") .testEval("'ogooglo'.rsplit('o')", "['', 'g', '', 'gl', '']") .testEval("'ogooglo'.rsplit('o', 1)", "['ogoogl', '']") .testEval("'ogooglo'.rsplit('o', 2)", "['ogo', 'gl', '']") .testEval("'ogooglo'.rsplit('o', 3)", "['og', '', 'gl', '']") .testEval("'ogooglo'.rsplit('o', 4)", "['', 'g', '', 'gl', '']") .testEval("'ogooglo'.rsplit('o', 5)", "['', 'g', '', 'gl', '']") .testEval("'google'.rsplit('google')", "['', '']") .testEval("'google'.rsplit('google', 1)", "['', '']") .testEval("'google'.rsplit('google', 2)", "['', '']"); } @Test public void testPyStringPartitionEasy() throws Exception { new BothModesTest() .testEval("'lawl'.partition('a')", "('l', 'a', 'wl')") .testEval("'lawl'.rpartition('a')", "('l', 'a', 'wl')"); } @Test public void testPyStringPartitionMultipleSep() throws Exception { new BothModesTest() .testEval("'google'.partition('o')", "('g', 'o', 'ogle')") .testEval("'google'.rpartition('o')", "('go', 'o', 'gle')") .testEval("'xxx'.partition('x')", "('', 'x', 'xx')") .testEval("'xxx'.rpartition('x')", "('xx', 'x', '')"); } @Test public void testPyStringPartitionEmptyInput() throws Exception { new BothModesTest() .testEval("''.partition('a')", "('', '', '')") .testEval("''.rpartition('a')", "('', '', '')"); } @Test public void testPyStringPartitionEmptySeparator() throws Exception { new BothModesTest() .testIfErrorContains("Empty separator", "'google'.partition('')") .testIfErrorContains("Empty separator", "'google'.rpartition('')"); } @Test public void testPyStringPartitionDefaultSep() throws Exception { new BothModesTest() .testEval("'hi this is a test'.partition()", "('hi', ' ', 'this is a test')") .testEval("'hi this is a test'.rpartition()", "('hi this is a', ' ', 'test')") .testEval("'google'.partition()", "('google', '', '')") .testEval("'google'.rpartition()", "('', '', 'google')"); } @Test public void testPyStringPartitionNoMatch() throws Exception { new BothModesTest() .testEval("'google'.partition('x')", "('google', '', '')") .testEval("'google'.rpartition('x')", "('', '', 'google')"); } @Test public void testPyStringPartitionWordBoundaries() throws Exception { new BothModesTest() .testEval("'goog'.partition('g')", "('', 'g', 'oog')") .testEval("'goog'.rpartition('g')", "('goo', 'g', '')") .testEval("'plex'.partition('p')", "('', 'p', 'lex')") .testEval("'plex'.rpartition('p')", "('', 'p', 'lex')") .testEval("'plex'.partition('x')", "('ple', 'x', '')") .testEval("'plex'.rpartition('x')", "('ple', 'x', '')"); } @Test public void testPyStringPartitionLongSep() throws Exception { new BothModesTest() .testEval("'google'.partition('oog')", "('g', 'oog', 'le')") .testEval("'google'.rpartition('oog')", "('g', 'oog', 'le')") .testEval( "'lolgooglolgooglolgooglol'.partition('goog')", "('lol', 'goog', 'lolgooglolgooglol')") .testEval( "'lolgooglolgooglolgooglol'.rpartition('goog')", "('lolgooglolgooglol', 'goog', 'lol')"); } @Test public void testPyStringPartitionCompleteString() throws Exception { new BothModesTest() .testEval("'google'.partition('google')", "('', 'google', '')") .testEval("'google'.rpartition('google')", "('', 'google', '')"); } @Test public void testPyStringTitle() throws Exception { new BothModesTest() .testStatement("'this is a very simple test'.title()", "This Is A Very Simple Test"); new BothModesTest() .testStatement("'Do We Keep Capital Letters?'.title()", "Do We Keep Capital Letters?"); new BothModesTest() .testStatement( "'this isn\\'t just an ol\\' apostrophe test'.title()", "This Isn'T Just An Ol' Apostrophe Test"); new BothModesTest() .testStatement( "'Let us test crazy characters: _bla.exe//foo:bla(test$class)'.title()", "Let Us Test Crazy Characters: _Bla.Exe//Foo:Bla(Test$Class)"); new BothModesTest().testStatement("'any germans here? äöü'.title()", "Any Germans Here? Äöü"); new BothModesTest() .testStatement( "'WE HAve tO lOWERCASE soMEthING heRE, AI?'.title()", "We Have To Lowercase Something Here, Ai?"); new BothModesTest() .testStatement("'wh4t ab0ut s0me numb3rs'.title()", "Wh4T Ab0Ut S0Me Numb3Rs"); } @Test public void testCapitalize() throws Exception { new BothModesTest() .testStatement("'hello world'.capitalize()", "Hello world") .testStatement("'HELLO WORLD'.capitalize()", "Hello world") .testStatement("''.capitalize()", "") .testStatement("'12 lower UPPER 34'.capitalize()", "12 lower upper 34"); } @Test public void testPyStringRfind() throws Exception { new BothModesTest() .testStatement("'banana'.rfind('na')", 4) .testStatement("'banana'.rfind('na', 3, 1)", -1) .testStatement("'aaaa'.rfind('a', 1, 1)", -1) .testStatement("'aaaa'.rfind('a', 1, 50)", 3) .testStatement("'aaaa'.rfind('aaaaa')", -1) .testStatement("'abababa'.rfind('ab', 1)", 4) .testStatement("'abababa'.rfind('ab', 0)", 4) .testStatement("'abababa'.rfind('ab', -1)", -1) .testStatement("'abababa'.rfind('ab', -2)", -1) .testStatement("'abababa'.rfind('ab', -3)", 4) .testStatement("'abababa'.rfind('ab', 0, 1)", -1) .testStatement("'abababa'.rfind('ab', 0, 2)", 0) .testStatement("'abababa'.rfind('ab', -1000)", 4) .testStatement("'abababa'.rfind('ab', 1000)", -1) .testStatement("''.rfind('a', 1)", -1); } @Test public void testPyStringFind() throws Exception { new BothModesTest() .testStatement("'banana'.find('na')", 2) .testStatement("'banana'.find('na', 3, 1)", -1) .testStatement("'aaaa'.find('a', 1, 1)", -1) .testStatement("'aaaa'.find('a', 1, 50)", 1) .testStatement("'aaaa'.find('aaaaa')", -1) .testStatement("'abababa'.find('ab', 1)", 2) .testStatement("'abababa'.find('ab', 0)", 0) .testStatement("'abababa'.find('ab', -1)", -1) .testStatement("'abababa'.find('ab', -2)", -1) .testStatement("'abababa'.find('ab', -3)", 4) .testStatement("'abababa'.find('ab', 0, 1)", -1) .testStatement("'abababa'.find('ab', 0, 2)", 0) .testStatement("'abababa'.find('ab', -1000)", 0) .testStatement("'abababa'.find('ab', 1000)", -1) .testStatement("''.find('a', 1)", -1); } @Test public void testPyStringIndex() throws Exception { new BothModesTest() .testStatement("'banana'.index('na')", 2) .testStatement("'abababa'.index('ab', 1)", 2) .testIfErrorContains("substring \"foo\" not found in \"banana\"", "'banana'.index('foo')"); } @Test public void testPyStringRIndex() throws Exception { new BothModesTest() .testStatement("'banana'.rindex('na')", 4) .testStatement("'abababa'.rindex('ab', 1)", 4) .testIfErrorContains("substring \"foo\" not found in \"banana\"", "'banana'.rindex('foo')"); } @Test public void testPyStringEndswith() throws Exception { new BothModesTest() .testStatement("'Apricot'.endswith('cot')", true) .testStatement("'a'.endswith('')", true) .testStatement("''.endswith('')", true) .testStatement("'Apricot'.endswith('co')", false) .testStatement("'Apricot'.endswith('co', -1)", false) .testStatement("'abcd'.endswith('c', -2, -1)", true) .testStatement("'abcd'.endswith('c', 1, 8)", false) .testStatement("'abcd'.endswith('d', 1, 8)", true); } @Test public void testPyStringStartswith() throws Exception { new BothModesTest() .testStatement("'Apricot'.startswith('Apr')", true) .testStatement("'Apricot'.startswith('A')", true) .testStatement("'Apricot'.startswith('')", true) .testStatement("'Apricot'.startswith('z')", false) .testStatement("''.startswith('')", true) .testStatement("''.startswith('a')", false); } @Test public void testPySubstring() throws Exception { new BothModesTest() .testStatement("'012345678'[0:-1]", "01234567") .testStatement("'012345678'[2:4]", "23") .testStatement("'012345678'[-5:-3]", "45") .testStatement("'012345678'[2:2]", "") .testStatement("'012345678'[2:]", "2345678") .testStatement("'012345678'[:3]", "012") .testStatement("'012345678'[-1:]", "8") .testStatement("'012345678'[:]", "012345678") .testStatement("'012345678'[-1:2]", "") .testStatement("'012345678'[4:2]", ""); } @Test public void testPyStringFormatEscaping() throws Exception { new BothModesTest() .testStatement("'{{}}'.format()", "{}") .testStatement("'{{}}'.format(42)", "{}") .testStatement("'{{ }}'.format()", "{ }") .testStatement("'{{ }}'.format(42)", "{ }") .testStatement("'{{{{}}}}'.format()", "{{}}") .testStatement("'{{{{}}}}'.format(42)", "{{}}") .testStatement("'{{0}}'.format(42)", "{0}") .testStatement("'{{}}'.format(42)", "{}") .testStatement("'{{{}}}'.format(42)", "{42}") .testStatement("'{{ '.format(42)", "{ ") .testStatement("' }}'.format(42)", " }") .testStatement("'{{ {}'.format(42)", "{ 42") .testStatement("'{} }}'.format(42)", "42 }") .testStatement("'{{0}}'.format(42)", "{0}") .testStatement("'{{{0}}}'.format(42)", "{42}") .testStatement("'{{ 0'.format(42)", "{ 0") .testStatement("'0 }}'.format(42)", "0 }") .testStatement("'{{ {0}'.format(42)", "{ 42") .testStatement("'{0} }}'.format(42)", "42 }") .testStatement("'{{test}}'.format(test = 42)", "{test}") .testStatement("'{{{test}}}'.format(test = 42)", "{42}") .testStatement("'{{ test'.format(test = 42)", "{ test") .testStatement("'test }}'.format(test = 42)", "test }") .testStatement("'{{ {test}'.format(test = 42)", "{ 42") .testStatement("'{test} }}'.format(test = 42)", "42 }") .testIfErrorContains("Found '}' without matching '{'", "'{{}'.format(1)") .testIfErrorContains("Found '}' without matching '{'", "'{}}'.format(1)"); } @Test public void testPyStringFormatManualPositionals() throws Exception { new BothModesTest() .testStatement( "'{0}, {1} {2} {3} test'.format('hi', 'this', 'is', 'a')", "hi, this is a test") .testStatement( "'{3}, {2} {1} {0} test'.format('a', 'is', 'this', 'hi')", "hi, this is a test") .testStatement( "'skip some {0}'.format('arguments', 'obsolete', 'deprecated')", "skip some arguments") .testStatement( "'{0} can be reused: {0}'.format('this', 'obsolete')", "this can be reused: this"); } @Test public void testPyStringFormatManualPositionalsErrors() throws Exception { new BothModesTest() .testIfErrorContains("No replacement found for index 0", "'{0}'.format()") .testIfErrorContains("No replacement found for index 1", "'{0} and {1}'.format('this')") .testIfErrorContains( "No replacement found for index 2", "'{0} and {2}'.format('this', 'that')") .testIfErrorContains( "No replacement found for index -1", "'{-0} and {-1}'.format('this', 'that')") .testIfErrorContains( "Invalid character ',' inside replacement field", "'{0,1} and {1}'.format('this', 'that')") .testIfErrorContains( "Invalid character '.' inside replacement field", "'{0.1} and {1}'.format('this', 'that')"); } @Test public void testPyStringFormatAutomaticPositionals() throws Exception { new BothModesTest() .testStatement("'{}, {} {} {} test'.format('hi', 'this', 'is', 'a')", "hi, this is a test") .testStatement( "'skip some {}'.format('arguments', 'obsolete', 'deprecated')", "skip some arguments"); } @Test public void testPyStringFormatAutomaticPositionalsError() throws Exception { new BothModesTest() .testIfErrorContains("No replacement found for index 0", "'{}'.format()") .testIfErrorContains("No replacement found for index 1", "'{} and {}'.format('this')"); } @Test public void testPyStringFormatMixedFields() throws Exception { new BothModesTest() .testStatement("'{test} and {}'.format(2, test = 1)", "1 and 2") .testStatement("'{test} and {0}'.format(2, test = 1)", "1 and 2") .testIfErrorContains( "non-keyword arg after keyword arg", "'{test} and {}'.format(test = 1, 2)") .testIfErrorContains( "non-keyword arg after keyword arg", "'{test} and {0}'.format(test = 1, 2)") .testIfErrorContains( "Cannot mix manual and automatic numbering of positional fields", "'{} and {1}'.format(1, 2)") .testIfErrorContains( "Cannot mix manual and automatic numbering of positional fields", "'{1} and {}'.format(1, 2)"); } @Test public void testPyStringFormatInvalidFields() throws Exception { for (char unsupported : new char[] {'.', '[', ']', ','}) { new BothModesTest() .testIfErrorContains( String.format("Invalid character '%c' inside replacement field", unsupported), String.format("'{test%ctest}'.format(test = 1)", unsupported)); } new BothModesTest() .testIfErrorContains("Nested replacement fields are not supported", "'{ {} }'.format(42)"); } @Test public void testPyStringFormat() throws Exception { new BothModesTest() .testStatement("'abc'.format()", "abc") .testStatement("'x{key}x'.format(key = 2)", "x2x") .testStatement("'x{key}x'.format(key = 'abc')", "xabcx") .testStatement("'{a}{b}{a}{b}'.format(a = 3, b = True)", "3True3True") .testStatement("'{a}{b}{a}{b}'.format(a = 3, b = True)", "3True3True") .testStatement("'{s1}{s2}'.format(s1 = ['a'], s2 = 'a')", "[\"a\"]a") .testIfErrorContains("Missing argument 'b'", "'{a}{b}'.format(a = 5)") .testStatement("'{a}'.format(a = '$')", "$") .testStatement("'{a}'.format(a = '$a')", "$a") .testStatement("'{a}$'.format(a = '$a')", "$a$"); // The test below is using **kwargs, which is not available in BUILD mode. new SkylarkTest().testStatement("'{(}'.format(**{'(': 2})", "2"); } @Test public void testReversedWithInvalidTypes() throws Exception { new BothModesTest() .testIfExactError("type 'NoneType' is not iterable", "reversed(None)") .testIfExactError("type 'int' is not iterable", "reversed(1)") .testIfExactError( "Argument to reversed() must be a sequence, not a dictionary.", "reversed({1: 3})"); new SkylarkTest() .testIfExactError( "Argument to reversed() must be a sequence, not a depset.", "reversed(depset([1]))"); } @Test public void testReversedWithLists() throws Exception { new BothModesTest() .testEval("reversed([])", "[]") .testEval("reversed([1])", "[1]") .testEval("reversed([1, 2, 3, 4, 5])", "[5, 4, 3, 2, 1]") .testEval("reversed([[1, 2], 3, 4, [5]])", "[[5], 4, 3, [1, 2]]") .testEval("reversed([1, 1, 1, 1, 2])", "[2, 1, 1, 1, 1]"); } @Test public void testReversedWithStrings() throws Exception { new BothModesTest() .testEval("reversed('')", "[]") .testEval("reversed('a')", "['a']") .testEval("reversed('abc')", "['c', 'b', 'a']") .testEval("reversed('__test ')", "[' ', ' ', 't', 's', 'e', 't', '_', '_']") .testEval("reversed('bbb')", "['b', 'b', 'b']"); } @Test public void testReversedNoSideEffects() throws Exception { new SkylarkTest() .testEval( "def foo():\n" + " x = ['a', 'b']\n" + " y = reversed(x)\n" + " y += ['c']\n" + " return x\n" + "foo()", "['a', 'b']"); } @Test public void testEquivalenceOfReversedAndSlice() throws Exception { String[] data = new String[] {"[]", "[1]", "[1, 2, 3]"}; for (String toBeReversed : data) { new BothModesTest() .testEval( String.format("reversed(%s)", toBeReversed), String.format("%s[::-1]", toBeReversed)); } } @Test public void testListSlice() throws Exception { new BothModesTest() .testEval("[0, 1, 2, 3][0:-1]", "[0, 1, 2]") .testEval("[0, 1, 2, 3, 4, 5][2:4]", "[2, 3]") .testEval("[0, 1, 2, 3, 4, 5][-2:-1]", "[4]") .testEval("[][1:2]", "[]") .testEval("[0, 1, 2, 3][-10:10]", "[0, 1, 2, 3]"); } @Test public void testListSlice_WrongType() throws Exception { new BothModesTest() .testIfExactError("slice start must be an integer, not 'a'", "'123'['a'::]") .testIfExactError("slice end must be an integer, not 'b'", "'123'[:'b':]"); } @Test public void testListSliceStep() throws Exception { new BothModesTest() .testEval("[1, 2, 3, 4, 5][::1]", "[1, 2, 3, 4, 5]") .testEval("[1, 2, 3, 4, 5][1::1]", "[2, 3, 4, 5]") .testEval("[1, 2, 3, 4, 5][:2:1]", "[1, 2]") .testEval("[1, 2, 3, 4, 5][1:3:1]", "[2, 3]") .testEval("[1, 2, 3, 4, 5][-4:-2:1]", "[2, 3]") .testEval("[1, 2, 3, 4, 5][-10:10:1]", "[1, 2, 3, 4, 5]") .testEval("[1, 2, 3, 4, 5][::42]", "[1]"); } @Test public void testListSliceStep_EmptyList() throws Exception { new BothModesTest().testEval("[][::1]", "[]").testEval("[][::-1]", "[]"); } @Test public void testListSliceStep_SkipValues() throws Exception { new BothModesTest() .testEval("[1, 2, 3, 4, 5, 6, 7][::3]", "[1, 4, 7]") .testEval("[1, 2, 3, 4, 5, 6, 7, 8, 9][1:7:3]", "[2, 5]"); } @Test public void testListSliceStep_Negative() throws Exception { new BothModesTest() .testEval("[1, 2, 3, 4, 5][::-1]", "[5, 4, 3, 2, 1]") .testEval("[1, 2, 3, 4, 5][4::-1]", "[5, 4, 3, 2, 1]") .testEval("[1, 2, 3, 4, 5][:0:-1]", "[5, 4, 3, 2]") .testEval("[1, 2, 3, 4, 5][3:1:-1]", "[4, 3]") .testEval("[1, 2, 3, 4, 5][::-2]", "[5, 3, 1]") .testEval("[1, 2, 3, 4, 5][::-10]", "[5]"); } @Test public void testListSlice_WrongOrder() throws Exception { new BothModesTest().testEval("[1, 2, 3][3:1:1]", "[]").testEval("[1, 2, 3][1:3:-1]", "[]"); } @Test public void testListSliceStep_InvalidStep() throws Exception { String msg = "slice step cannot be zero"; new BothModesTest() .testIfExactError(msg, "[1, 2, 3][::0]") .testIfExactError(msg, "[1, 2, 3][1::0]") .testIfExactError(msg, "[1, 2, 3][:3:0]") .testIfExactError(msg, "[1, 2, 3][1:3:0]"); } @Test public void testTupleSlice() throws Exception { // Not as comprehensive as the tests for slicing lists since the underlying mechanism is the // same. new BothModesTest() .testEval("()[1:2]", "()") .testEval("()[::1]", "()") .testEval("(0, 1, 2, 3)[0:-1]", "(0, 1, 2)") .testEval("(0, 1, 2, 3, 4, 5)[2:4]", "(2, 3)") .testEval("(0, 1, 2, 3)[-10:10]", "(0, 1, 2, 3)") .testEval("(1, 2, 3, 4, 5)[-10:10:1]", "(1, 2, 3, 4, 5)") .testEval("(1, 2, 3, 4, 5, 6, 7, 8, 9)[1:7:3]", "(2, 5)") .testEval("(1, 2, 3, 4, 5)[::-1]", "(5, 4, 3, 2, 1)") .testEval("(1, 2, 3, 4, 5)[3:1:-1]", "(4, 3)") .testEval("(1, 2, 3, 4, 5)[::-2]", "(5, 3, 1)") .testEval("(1, 2, 3, 4, 5)[::-10]", "(5,)") .testIfExactError("slice step cannot be zero", "(1, 2, 3)[1::0]"); } @Test public void testListSort() throws Exception { new BothModesTest() .testEval("sorted([0,1,2,3])", "[0, 1, 2, 3]") .testEval("sorted([])", "[]") .testEval("sorted([3, 2, 1, 0])", "[0, 1, 2, 3]") .testEval("sorted([[1], [], [2], [1, 2]])", "[[], [1], [1, 2], [2]]") .testEval("sorted([True, False, True])", "[False, True, True]") .testEval("sorted(['a','x','b','z'])", "[\"a\", \"b\", \"x\", \"z\"]") .testEval("sorted({1: True, 5: True, 4: False})", "[1, 4, 5]") .testEval("sorted(depset([1, 5, 4]))", "[1, 4, 5]") .testIfExactError("Cannot compare function with function", "sorted([sorted, sorted])"); } @Test public void testDictionaryCopy() throws Exception { new BothModesTest() .setUp("x = {1 : 2}", "y = dict(x)") .testEval("x[1] == 2 and y[1] == 2", "True"); } @Test public void testDictionaryCopyKeyCollision() throws Exception { new BothModesTest() .setUp("x = {'test' : 2}", "y = dict(x, test = 3)") .testEval("y['test']", "3"); } @Test public void testDictionaryWithMultipleKeys() throws Exception { new BothModesTest().testStatement("{0: 'a', 1: 'b', 0: 'c'}[0]", "c"); } @Test public void testDictionaryKeyNotFound() throws Exception { new BothModesTest() .testIfErrorContains("key \"0\" not found in dictionary", "{}['0']") .testIfErrorContains("key 0 not found in dictionary", "{'0': 1, 2: 3, 4: 5}[0]"); } @Test public void testListAccessBadIndex() throws Exception { new BothModesTest() .testIfErrorContains("indices must be integers, not string", "[[1], [2]]['a']"); } @Test public void testDictionaryAccess() throws Exception { new BothModesTest() .testEval("{1: ['foo']}[1]", "['foo']") .testStatement("{'4': 8}['4']", 8) .testStatement("{'a': 'aa', 'b': 'bb', 'c': 'cc'}['b']", "bb"); } @Test public void testDictionaryVariableAccess() throws Exception { new BothModesTest().setUp("d = {'a' : 1}", "a = d['a']\n").testLookup("a", 1); } @Test public void testStringIndexing() throws Exception { new BothModesTest() .testStatement("'somestring'[0]", "s") .testStatement("'somestring'[1]", "o") .testStatement("'somestring'[4]", "s") .testStatement("'somestring'[9]", "g") .testStatement("'somestring'[-1]", "g") .testStatement("'somestring'[-2]", "n") .testStatement("'somestring'[-10]", "s"); } @Test public void testStringIndexingOutOfRange() throws Exception { new BothModesTest() .testIfErrorContains("index out of range", "'abcdef'[10]") .testIfErrorContains("index out of range", "'abcdef'[-11]") .testIfErrorContains("index out of range", "'abcdef'[42]"); } @Test public void testStringSlice() throws Exception { new BothModesTest() .testStatement("'0123'[0:-1]", "012") .testStatement("'012345'[2:4]", "23") .testStatement("'012345'[-2:-1]", "4") .testStatement("''[1:2]", "") .testStatement("'012'[1:0]", "") .testStatement("'0123'[-10:10]", "0123"); } @Test public void testStringSlice_WrongType() throws Exception { new BothModesTest() .testIfExactError("slice start must be an integer, not 'a'", "'123'['a'::]") .testIfExactError("slice end must be an integer, not 'b'", "'123'[:'b':]"); } @Test public void testStringSliceStep() throws Exception { new BothModesTest() .testStatement("'01234'[::1]", "01234") .testStatement("'01234'[1::1]", "1234") .testStatement("'01234'[:2:1]", "01") .testStatement("'01234'[1:3:1]", "12") .testStatement("'01234'[-4:-2:1]", "12") .testStatement("'01234'[-10:10:1]", "01234") .testStatement("'01234'[::42]", "0"); } @Test public void testStringSliceStep_EmptyString() throws Exception { new BothModesTest().testStatement("''[::1]", "").testStatement("''[::-1]", ""); } @Test public void testStringSliceStep_SkipValues() throws Exception { new BothModesTest() .testStatement("'0123456'[::3]", "036") .testStatement("'01234567'[1:7:3]", "14"); } @Test public void testStringSliceStep_Negative() throws Exception { new BothModesTest() .testStatement("'01234'[::-1]", "43210") .testStatement("'01234'[4::-1]", "43210") .testStatement("'01234'[:0:-1]", "4321") .testStatement("'01234'[3:1:-1]", "32") .testStatement("'01234'[::-2]", "420") .testStatement("'01234'[::-10]", "4"); } @Test public void testStringSliceStep_WrongOrder() throws Exception { new BothModesTest().testStatement("'123'[3:1:1]", "").testStatement("'123'[1:3:-1]", ""); } @Test public void testStringSliceStep_InvalidStep() throws Exception { String msg = "slice step cannot be zero"; new BothModesTest() .testIfExactError(msg, "'123'[::0]") .testIfExactError(msg, "'123'[1::0]") .testIfExactError(msg, "'123'[:3:0]") .testIfExactError(msg, "'123'[1:3:0]"); } @Test public void testDictionaryCreation() throws Exception { String expected = "{'a': 1, 'b': 2, 'c': 3}"; new BothModesTest() .testEval("dict([('a', 1), ('b', 2), ('c', 3)])", expected) .testEval("dict(a = 1, b = 2, c = 3)", expected) .testEval("dict([('a', 1)], b = 2, c = 3)", expected); } @Test public void testDictionaryCreationInnerLists() throws Exception { new BothModesTest().testEval("dict([[1, 2], [3, 4]], a = 5)", "{1: 2, 3: 4, 'a': 5}"); } @Test public void testDictionaryCreationEmpty() throws Exception { new BothModesTest().testEval("dict()", "{}").testEval("dict([])", "{}"); } @Test public void testDictionaryCreationDifferentKeyTypes() throws Exception { String expected = "{'a': 1, 2: 3}"; new BothModesTest() .testEval("dict([('a', 1), (2, 3)])", expected) .testEval("dict([(2, 3)], a = 1)", expected); } @Test public void testDictionaryCreationKeyCollision() throws Exception { String expected = "{'a': 1, 'b': 2, 'c': 3}"; new BothModesTest() .testEval("dict([('a', 42), ('b', 2), ('a', 1), ('c', 3)])", expected) .testEval("dict([('a', 42)], a = 1, b = 2, c = 3)", expected); new SkylarkTest().testEval("dict([('a', 42)], **{'a': 1, 'b': 2, 'c': 3})", expected); } @Test public void testDictionaryCreationInvalidPositional() throws Exception { new BothModesTest() .testIfErrorContains( "expected value of type 'list(object)' for parameter args in dict(), " + "but got \"a\" (string)", "dict('a')") .testIfErrorContains("cannot convert item #0 to a sequence", "dict(['a'])") .testIfErrorContains("cannot convert item #0 to a sequence", "dict([('a')])") .testIfErrorContains("too many (3) positional arguments", "dict((3,4), (3,2), (1,2))") .testIfErrorContains( "item #0 has length 3, but exactly two elements are required", "dict([('a', 'b', 'c')])"); } @Test public void testDictionaryValues() throws Exception { new BothModesTest() .testEval("{1: 'foo'}.values()", "['foo']") .testEval("{}.values()", "[]") .testEval("{True: 3, False: 5}.values()", "[3, 5]") .testEval("{'a': 5, 'c': 2, 'b': 4, 'd': 3}.values()", "[5, 2, 4, 3]"); // sorted by keys } @Test public void testDictionaryKeys() throws Exception { new BothModesTest() .testEval("{1: 'foo'}.keys()", "[1]") .testEval("{}.keys()", "[]") .testEval("{True: 3, False: 5}.keys()", "[True, False]") .testEval( "{1:'a', 2:'b', 6:'c', 0:'d', 5:'e', 4:'f', 3:'g'}.keys()", "[1, 2, 6, 0, 5, 4, 3]"); } @Test public void testDictionaryGet() throws Exception { new BuildTest() .testStatement("{1: 'foo'}.get(1)", "foo") .testStatement("{1: 'foo'}.get(2)", Runtime.NONE) .testStatement("{1: 'foo'}.get(2, 'a')", "a") .testStatement("{1: 'foo'}.get(2, default='a')", "a") .testStatement("{1: 'foo'}.get(2, default=None)", Runtime.NONE); } @Test public void testDictionaryItems() throws Exception { new BothModesTest() .testEval("{'a': 'foo'}.items()", "[('a', 'foo')]") .testEval("{}.items()", "[]") .testEval("{1: 3, 2: 5}.items()", "[(1, 3), (2, 5)]") .testEval("{'a': 5, 'c': 2, 'b': 4}.items()", "[('a', 5), ('c', 2), ('b', 4)]"); } @Test public void testDictionaryClear() throws Exception { new SkylarkTest() .testEval( "d = {1: 'foo', 2: 'bar', 3: 'baz'}\n" + "if len(d) != 3: fail('clear 1')\n" + "if d.clear() != None: fail('clear 2')\n" + "d", "{}"); } @Test public void testDictionaryPop() throws Exception { new SkylarkTest() .testIfErrorContains( "KeyError: 1", "d = {1: 'foo', 2: 'bar', 3: 'baz'}\n" + "if len(d) != 3: fail('pop 1')\n" + "if d.pop(2) != 'bar': fail('pop 2')\n" + "if d.pop(3, 'quux') != 'baz': fail('pop 3a')\n" + "if d.pop(3, 'quux') != 'quux': fail('pop 3b')\n" + "if d.pop(1) != 'foo': fail('pop 1')\n" + "if d != {}: fail('pop 0')\n" + "d.pop(1)"); } @Test public void testDictionaryPopItem() throws Exception { new SkylarkTest() .testIfErrorContains( "popitem(): dictionary is empty", "d = {2: 'bar', 3: 'baz', 1: 'foo'}\n" + "if len(d) != 3: fail('popitem 0')\n" + "if d.popitem() != (2, 'bar'): fail('popitem 2')\n" + "if d.popitem() != (3, 'baz'): fail('popitem 3')\n" + "if d.popitem() != (1, 'foo'): fail('popitem 1')\n" + "if d != {}: fail('popitem 4')\n" + "d.popitem()"); } @Test public void testDictionaryUpdate() throws Exception { new BothModesTest() .setUp("foo = {'a': 2}") .testEval("foo.update({'b': 4}); foo", "{'a': 2, 'b': 4}"); new BothModesTest() .setUp("foo = {'a': 2}") .testEval("foo.update({'a': 3, 'b': 4}); foo", "{'a': 3, 'b': 4}"); } @Test public void testDictionarySetDefault() throws Exception { new SkylarkTest() .testEval( "d = {2: 'bar', 1: 'foo'}\n" + "if len(d) != 2: fail('setdefault 0')\n" + "if d.setdefault(1, 'a') != 'foo': fail('setdefault 1')\n" + "if d.setdefault(2) != 'bar': fail('setdefault 2')\n" + "if d.setdefault(3) != None: fail('setdefault 3')\n" + "if d.setdefault(4, 'b') != 'b': fail('setdefault 4')\n" + "d", "{1: 'foo', 2: 'bar', 3: None, 4: 'b'}"); } @Test public void testListIndexMethod() throws Exception { new BothModesTest() .testStatement("['a', 'b', 'c'].index('a')", 0) .testStatement("['a', 'b', 'c'].index('b')", 1) .testStatement("['a', 'b', 'c'].index('c')", 2) .testStatement("[2, 4, 6].index(4)", 1) .testStatement("[2, 4, 6].index(4)", 1) .testStatement("[0, 1, [1]].index([1])", 2) .testIfErrorContains("item \"a\" not found in list", "[1, 2].index('a')") .testIfErrorContains("item 0 not found in list", "[].index(0)"); } @Test public void testListIndex() throws Exception { new BothModesTest() .testStatement("['a', 'b', 'c', 'd'][0]", "a") .testStatement("['a', 'b', 'c', 'd'][1]", "b") .testStatement("['a', 'b', 'c', 'd'][-1]", "d") .testStatement("['a', 'b', 'c', 'd'][-2]", "c") .testStatement("[0, 1, 2][-3]", 0) .testStatement("[0, 1, 2][-2]", 1) .testStatement("[0, 1, 2][-1]", 2) .testStatement("[0, 1, 2][0]", 0); } @Test public void testListIndexOutOfRange() throws Exception { new BothModesTest() .testIfErrorContains( "index out of range (index is 3, but sequence has 3 elements)", "[0, 1, 2][3]") .testIfErrorContains( "index out of range (index is -4, but sequence has 3 elements)", "[0, 1, 2][-4]") .testIfErrorContains( "index out of range (index is -2, but sequence has 1 elements)", "[0][-2]") .testIfErrorContains( "index out of range (index is 1, but sequence has 1 elements)", "[0][1]") .testIfErrorContains( "index out of range (index is 1, but sequence has 0 elements)", "[][1]"); } @Test public void testHash() throws Exception { // We specify the same string hashing algorithm as String.hashCode(). new SkylarkTest() .testStatement("hash('skylark')", "skylark".hashCode()) .testStatement("hash('google')", "google".hashCode()) .testIfErrorContains( "method hash(value: string) is not applicable for arguments (NoneType): " + "'value' is 'NoneType', but should be 'string'", "hash(None)"); } @Test public void testRange() throws Exception { new BothModesTest() .testStatement("str(range(5))", "[0, 1, 2, 3, 4]") .testStatement("str(range(0))", "[]") .testStatement("str(range(1))", "[0]") .testStatement("str(range(-2))", "[]") .testStatement("str(range(-3, 2))", "[-3, -2, -1, 0, 1]") .testStatement("str(range(3, 2))", "[]") .testStatement("str(range(3, 3))", "[]") .testStatement("str(range(3, 4))", "[3]") .testStatement("str(range(3, 5))", "[3, 4]") .testStatement("str(range(-3, 5, 2))", "[-3, -1, 1, 3]") .testStatement("str(range(-3, 6, 2))", "[-3, -1, 1, 3, 5]") .testStatement("str(range(5, 0, -1))", "[5, 4, 3, 2, 1]") .testStatement("str(range(5, 0, -10))", "[5]") .testStatement("str(range(0, -3, -2))", "[0, -2]") .testIfErrorContains("step cannot be 0", "range(2, 3, 0)"); } @Test public void testEnumerate() throws Exception { new BothModesTest() .testStatement("str(enumerate([]))", "[]") .testStatement("str(enumerate([5]))", "[(0, 5)]") .testStatement("str(enumerate([5, 3]))", "[(0, 5), (1, 3)]") .testStatement("str(enumerate(['a', 'b', 'c']))", "[(0, \"a\"), (1, \"b\"), (2, \"c\")]") .testStatement("str(enumerate(['a']) + [(1, 'b')])", "[(0, \"a\"), (1, \"b\")]"); } @Test public void testEnumerateBadArg() throws Exception { new BothModesTest() .testIfErrorContains( "method enumerate(list: sequence) is not applicable for arguments (string): " + "'list' is 'string', but should be 'sequence'", "enumerate('a')"); } @Test public void testPyListAppend() throws Exception { new BuildTest() .setUp("FOO = ['a', 'b']", "FOO.insert(0, 'c')") .testLookup("FOO", MutableList.of(env, "c", "a", "b")) .setUp("FOO.insert(1, 'd')") .testLookup("FOO", MutableList.of(env, "c", "d", "a", "b")) .setUp("FOO.insert(4, 'e')") .testLookup("FOO", MutableList.of(env, "c", "d", "a", "b", "e")) .setUp("FOO.insert(-10, 'f')") .testLookup("FOO", MutableList.of(env, "f", "c", "d", "a", "b", "e")) .setUp("FOO.insert(10, 'g')") .testLookup("FOO", MutableList.of(env, "f", "c", "d", "a", "b", "e", "g")) .testIfErrorContains("type 'tuple' has no method insert(int)", "(1, 2).insert(3)"); } @Test public void testPyListInsert() throws Exception { new BuildTest() .setUp("FOO = ['a', 'b']", "FOO.append('c')") .testLookup("FOO", MutableList.of(env, "a", "b", "c")) .testIfErrorContains("type 'tuple' has no method append(int)", "(1, 2).append(3)"); } @Test public void testPyListExtend() throws Exception { new BuildTest() .setUp("FOO = ['a', 'b']", "FOO.extend(['c', 'd'])", "FOO.extend(('e', 'f'))") .testLookup("FOO", MutableList.of(env, "a", "b", "c", "d", "e", "f")) .testIfErrorContains("type 'tuple' has no method extend(list)", "(1, 2).extend([3, 4])") .testIfErrorContains( "method list.extend(items: sequence) is not applicable for arguments " + "(int): 'items' is 'int', but should be 'sequence'", "[1, 2].extend(3)"); } @Test public void testListRemove() throws Exception { new BothModesTest() .setUp("foo = ['a', 'b', 'c', 'b']", "foo.remove('b')") .testLookup("foo", MutableList.of(env, "a", "c", "b")) .setUp("foo.remove('c')") .testLookup("foo", MutableList.of(env, "a", "b")) .setUp("foo.remove('a')") .testLookup("foo", MutableList.of(env, "b")) .setUp("foo.remove('b')") .testLookup("foo", MutableList.of(env)) .testIfErrorContains("item 3 not found in list", "[1, 2].remove(3)"); new BothModesTest() .testIfErrorContains("type 'tuple' has no method remove(int)", "(1, 2).remove(3)"); } @Test public void testListPop() throws Exception { new BothModesTest() .setUp("li = [2, 3, 4]; ret = li.pop()") .testLookup("li", MutableList.of(env, 2, 3)) .testLookup("ret", 4); new BothModesTest() .setUp("li = [2, 3, 4]; ret = li.pop(-2)") .testLookup("li", MutableList.of(env, 2, 4)) .testLookup("ret", 3); new BothModesTest() .setUp("li = [2, 3, 4]; ret = li.pop(1)") .testLookup("li", MutableList.of(env, 2, 4)) .testLookup("ret", 3); new BothModesTest() .testIfErrorContains( "index out of range (index is 3, but sequence has 2 elements)", "[1, 2].pop(3)"); new BothModesTest().testIfErrorContains("type 'tuple' has no method pop()", "(1, 2).pop()"); } @Test public void testReassignmentOfPrimitivesNotForbiddenByCoreLanguage() throws Exception { new BuildTest() .setUp("cc_binary = (['hello.cc'])") .testIfErrorContains( "'list' object is not callable", "cc_binary(name = 'hello', srcs=['hello.cc'], malloc = '//base:system_malloc')"); } @Test public void testLenOnString() throws Exception { new BothModesTest().testStatement("len('abc')", 3); } @Test public void testLenOnList() throws Exception { new BothModesTest().testStatement("len([1,2,3])", 3); } @Test public void testLenOnDict() throws Exception { new BothModesTest().testStatement("len({'a' : 1, 'b' : 2})", 2); } @Test public void testLenOnBadType() throws Exception { new BothModesTest().testIfErrorContains("int is not iterable", "len(1)"); } @Test public void testIndexOnFunction() throws Exception { new BothModesTest() .testIfErrorContains("type 'function' has no operator [](int)", "len[1]") .testIfErrorContains("type 'function' has no operator [:](int, int, int)", "len[1:4]"); } @Test public void testBool() throws Exception { new BothModesTest() .testStatement("bool(1)", Boolean.TRUE) .testStatement("bool(0)", Boolean.FALSE) .testStatement("bool([1, 2])", Boolean.TRUE) .testStatement("bool([])", Boolean.FALSE) .testStatement("bool(None)", Boolean.FALSE); } @Test public void testStr() throws Exception { new BothModesTest() .testStatement("str(1)", "1") .testStatement("str(-2)", "-2") .testStatement("str([1, 2])", "[1, 2]") .testStatement("str(True)", "True") .testStatement("str(False)", "False") .testStatement("str(None)", "None") .testStatement("str(str)", ""); } @Test public void testInt() throws Exception { new BothModesTest() .testStatement("int('1')", 1) .testStatement("int('-1234')", -1234) .testIfErrorContains("invalid literal for int() with base 10: \"1.5\"", "int('1.5')") .testIfErrorContains("invalid literal for int() with base 10: \"ab\"", "int('ab')") .testStatement("int(42)", 42) .testStatement("int(-1)", -1) .testStatement("int(True)", 1) .testStatement("int(False)", 0) .testIfErrorContains("None is not of type string or int or bool", "int(None)"); } @Test public void testIntWithBase() throws Exception { new BothModesTest() .testStatement("int('11', 2)", 3) .testStatement("int('11', 9)", 10) .testStatement("int('AF', 16)", 175) .testStatement("int('11', 36)", 37) .testStatement("int('az', 36)", 395); } @Test public void testIntWithBase_InvalidBase() throws Exception { new BothModesTest() .testIfExactError("invalid literal for int() with base 3: \"123\"", "int('123', 3)") .testIfExactError("invalid literal for int() with base 15: \"FF\"", "int('FF', 15)") .testIfExactError("int() base must be >= 2 and <= 36", "int('123', -1)") .testIfExactError("int() base must be >= 2 and <= 36", "int('123', 1)") .testIfExactError("int() base must be >= 2 and <= 36", "int('123', 37)"); } @Test public void testIntWithBase_Prefix() throws Exception { new BothModesTest() .testStatement("int('0b11', 0)", 3) .testStatement("int('0B11', 2)", 3) .testStatement("int('0o11', 0)", 9) .testStatement("int('0O11', 8)", 9) .testStatement("int('0XFF', 0)", 255) .testStatement("int('0xFF', 16)", 255) .testIfExactError("invalid literal for int() with base 8: \"0xFF\"", "int('0xFF', 8)"); } @Test public void testIntWithBase_NoString() throws Exception { new BothModesTest() .testIfExactError("int() can't convert non-string with explicit base", "int(True, 2)") .testIfExactError("int() can't convert non-string with explicit base", "int(1, 2)"); } @Test public void testStrFunction() throws Exception { new SkylarkTest().testStatement("def foo(x): return x\nstr(foo)", ""); } @Test public void testType() throws Exception { new SkylarkTest() .testStatement("type(1)", "int") .testStatement("type('a')", "string") .testStatement("type([1, 2])", "list") .testStatement("type((1, 2))", "tuple") .testStatement("type(True)", "bool") .testStatement("type(None)", "NoneType") .testStatement("type(str)", "function"); } // TODO(bazel-team): Move this into a new BazelLibraryTest.java file, or at least out of // MethodLibraryTest.java. @Test public void testSelectFunction() throws Exception { enableSkylarkMode(); eval("a = select({'a': 1})"); SelectorList result = (SelectorList) lookup("a"); assertThat(((SelectorValue) Iterables.getOnlyElement(result.getElements())).getDictionary()) .isEqualTo(ImmutableMap.of("a", 1)); } @Test public void testCountFunction() throws Exception { new BothModesTest() .testStatement("'abc'.count('')", 4) .testStatement("'abc'.count('a')", 1) .testStatement("'abc'.count('b')", 1) .testStatement("'abc'.count('c')", 1) .testStatement("'abbc'.count('b')", 2) .testStatement("'aba'.count('a')", 2) .testStatement("'aaa'.count('aa')", 1) .testStatement("'aaaa'.count('aa')", 2) .testStatement("'abc'.count('a', 0)", 1) .testStatement("'abc'.count('a', 1)", 0) .testStatement("'abc'.count('c', 0, 3)", 1) .testStatement("'abc'.count('c', 0, 2)", 0) .testStatement("'abc'.count('a', -1)", 0) .testStatement("'abc'.count('c', -1)", 1) .testStatement("'abc'.count('c', 0, 5)", 1) .testStatement("'abc'.count('c', 0, -1)", 0) .testStatement("'abc'.count('a', 0, -1)", 1); } @Test public void testZipFunction() throws Exception { new BothModesTest() .testStatement("str(zip())", "[]") .testStatement("str(zip([1, 2]))", "[(1,), (2,)]") .testStatement("str(zip([1, 2], ['a', 'b']))", "[(1, \"a\"), (2, \"b\")]") .testStatement("str(zip([1, 2, 3], ['a', 'b']))", "[(1, \"a\"), (2, \"b\")]") .testStatement("str(zip([1], [2], [3]))", "[(1, 2, 3)]") .testStatement("str(zip([1], {2: 'a'}))", "[(1, 2)]") .testStatement("str(zip([1], []))", "[]") .testIfErrorContains("type 'int' is not iterable", "zip(123)") .testIfErrorContains("type 'int' is not iterable", "zip([1], 1)") .testStatement("str(zip([1], depset([2])))", "[(1, 2)]"); } @Test public void testIsAlphaFunction() throws Exception { new BothModesTest() .testStatement("''.isalpha()", false) .testStatement("'abz'.isalpha()", true) .testStatement("'a1'.isalpha()", false) .testStatement("'a '.isalpha()", false) .testStatement("'A'.isalpha()", true) .testStatement("'AbZ'.isalpha()", true); } /** * Assert that lstrip(), rstrip(), and strip() produce the expected result for a given input * string and chars argument. If chars is null no argument is passed. */ private void checkStrip( String input, Object chars, String expLeft, String expRight, String expBoth) throws Exception { if (chars == null) { new BothModesTest() .update("s", input) .testStatement("s.lstrip()", expLeft) .testStatement("s.rstrip()", expRight) .testStatement("s.strip()", expBoth); } else { new BothModesTest() .update("s", input) .update("chars", chars) .testStatement("s.lstrip(chars)", expLeft) .testStatement("s.rstrip(chars)", expRight) .testStatement("s.strip(chars)", expBoth); } } @Test public void testStrip() throws Exception { // Strip nothing. checkStrip("a b c", "", "a b c", "a b c", "a b c"); checkStrip(" a b c ", "", " a b c ", " a b c ", " a b c "); // Normal case, found and not found. checkStrip("abcba", "ba", "cba", "abc", "c"); checkStrip("abc", "xyz", "abc", "abc", "abc"); // Default whitespace. checkStrip(" a b c ", null, "a b c ", " a b c", "a b c"); checkStrip(" a b c ", Runtime.NONE, "a b c ", " a b c", "a b c"); // Default whitespace with full range of Latin-1 whitespace chars. String whitespace = "\u0009\n\u000B\u000C\r\u001C\u001D\u001E\u001F\u0020\u0085\u00A0"; checkStrip( whitespace + "a" + whitespace, null, "a" + whitespace, whitespace + "a", "a"); checkStrip( whitespace + "a" + whitespace, Runtime.NONE, "a" + whitespace, whitespace + "a", "a"); // Empty cases. checkStrip("", "", "", "", ""); checkStrip("abc", "abc", "", "", ""); checkStrip("", "xyz", "", "", ""); checkStrip("", null, "", "", ""); } @Test public void testFail() throws Exception { new SkylarkTest() .testIfErrorContains("abc", "fail('abc')") .testIfErrorContains("18", "fail(18)"); } @Test public void testTupleCoercion() throws Exception { new BothModesTest() .testStatement("tuple([1, 2]) == (1, 2)", true) .testStatement("tuple(depset([1, 2])) == (1, 2)", true) // Depends on current implementation of dict .testStatement("tuple({1: 'foo', 2: 'bar'}) == (1, 2)", true); } }