jprdonnelly/impala
1
0
Fork 0
mirror of https://github.com/apache/impala.git synced 2025-12-25 02:03:09 -05:00
Code Issues Projects Releases Wiki Activity

Reenabled -1 to indicate no memory limit since CM uses that.

Browse Source
This commit is contained in:
Alex Behm
2013-04-01 16:15:05 -07:00
committed by Henry Robinson
parent ffb41b36be
commit fc310fadab
5 changed files with 20 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tests/query_test/test_query_mem_limit.py
View File

@@ -38,13 +38,18 @@ class TestQueryMemLimit(ImpalaTestSuite):
# dynamically, even if it is a rough approximation.
# A mem_limit is expressed in bytes, with values <= 0 signifying no cap.
# These values are either really small, unlimited, or have a really large cap.
SYS_MAXINT = str(sys.maxint)
MEM_LIMITS = ["-1", "0", "1", "10", "100", "1000", "10000", SYS_MAXINT,
SYS_MAXINT + "b", SYS_MAXINT + "B",
SYS_MAXINT + "m", SYS_MAXINT + "M",
SYS_MAXINT + "g", SYS_MAXINT + "G",
MAXINT_BYTES = str(sys.maxint)
MAXINT_MB = str(sys.maxint/(1024*1024))
MAXINT_GB = str(sys.maxint/(1024*1024*1024))
# We expect the tests with MAXINT_* using valid units [bmg] to succeed.
PASS_REGEX = re.compile("(%s|%s|%s)[bmg]?$" % (MAXINT_BYTES, MAXINT_MB, MAXINT_GB),
re.I)
MEM_LIMITS = ["-1", "0", "1", "10", "100", "1000", "10000", MAXINT_BYTES,
MAXINT_BYTES + "b", MAXINT_BYTES + "B",
MAXINT_MB + "m", MAXINT_MB + "M",
MAXINT_GB + "g", MAXINT_GB + "G",
# invalid per-query memory limits
"xyz", "100%", SYS_MAXINT + "k", "k" + SYS_MAXINT]
"-1234", "-3.14", "xyz", "100%", MAXINT_BYTES + "k", "k" + MAXINT_BYTES]
@classmethod
def get_workload(self):
@@ -74,7 +79,7 @@ class TestQueryMemLimit(ImpalaTestSuite):
exec_options['mem_limit'] = mem_limit
query = vector.get_value('query')
table_format = vector.get_value('table_format')
if mem_limit == "0" or re.match(self.SYS_MAXINT + "[bBmMgG]?$", mem_limit):
if mem_limit in["0", "-1"] or self.PASS_REGEX.match(mem_limit):
# should succeed
self.__exec_query(query, exec_options, True, table_format)
else: