<commit_before>#!/usr/bin/env python3

# Copyright 2013 <NAME>

# 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 sys, os, subprocess, time, datetime, pickle, multiprocessing
import concurrent.futures as conc
from optparse import OptionParser

tests_failed = False

parser = OptionParser()
parser.add_option('--wrapper', default=None, dest='wrapper',
                  help='wrapper to run tests with (e.g. valgrind)')

class TestRun():
    def __init__(self, res, duration, stdo, stde):
        self.res = res
        self.duration = duration
        self.stdo = stdo
        self.stde = stde

def write_log(logfile, test_name, result_str, stdo, stde):
    logfile.write(result_str + '\n\n')
    logfile.write('--- "%s" stdout ---\n' % test_name)
    logfile.write(stdo)
    logfile.write('\n--- "%s" stderr ---\n' % test_name)
    logfile.write(stde)
    logfile.write('\n-------\n\n')

def run_single_test(wrap, test):
    global tests_failed
    if test.is_cross:
        if test.exe_runner is None:
            # 'Can not run test on cross compiled executable 
            # because there is no execute wrapper.
            cmd = None
        else:
            cmd = [test.exe_runner, test.fname]
    else:
        cmd = [test.fname]
    if cmd is None:
        res = 'SKIP'
        duration = 0.0
        stdo = 'Not run because can not execute cross compiled binaries.'
        stde = ''
    else:
        cmd = wrap + cmd + test.cmd_args
        starttime = time.time()
        p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE,
                             env=test.env)
        (stdo, stde) = p.communicate()
        endtime = time.time()
        duration = endtime - starttime
        stdo = stdo.decode()
        stde = stde.decode()
        if p.returncode == 0:
            res = 'OK'
        else:
            res = 'FAIL'
            tests_failed = True
    return TestRun(res, duration, stdo, stde)

def print_stats(numlen, tests, name, result, i, logfile):
    startpad = ' '*(numlen - len('%d' % (i+1)))
    num = '%s%d/%d' % (startpad, i+1, len(tests))
    padding1 = ' '*(40-len(name))
    padding2 = ' '*(5-len(result.res))
    result_str = '%s %s%s%s%s(%5.2f s)' % \
        (num, name, padding1, result.res, padding2, result.duration)<commit_msg>Pass parent envvars to test executables.<commit_after>#!/usr/bin/env python3

# Copyright 2013 <NAME>

# 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 sys, os, subprocess, time, datetime, pickle, multiprocessing
import concurrent.futures as conc
from optparse import OptionParser

tests_failed = False

parser = OptionParser()
parser.add_option('--wrapper', default=None, dest='wrapper',
                  help='wrapper to run tests with (e.g. valgrind)')

class TestRun():
    def __init__(self, res, duration, stdo, stde):
        self.res = res
        self.duration = duration
        self.stdo = stdo
        self.stde = stde

def write_log(logfile, test_name, result_str, stdo, stde):
    logfile.write(result_str + '\n\n')
    logfile.write('--- "%s" stdout ---\n' % test_name)
    logfile.write(stdo)
    logfile.write('\n--- "%s" stderr ---\n' % test_name)
    logfile.write(stde)
    logfile.write('\n-------\n\n')

def run_single_test(wrap, test):
    global tests_failed
    if test.is_cross:
        if test.exe_runner is None:
            # 'Can not run test on cross compiled executable 
            # because there is no execute wrapper.
            cmd = None
        else:
            cmd = [test.exe_runner, test.fname]
    else:
        cmd = [test.fname]
    if cmd is None:
        res = 'SKIP'
        duration = 0.0
        stdo = 'Not run because can not execute cross compiled binaries.'
        stde = ''
    else:
        cmd = wrap + cmd + test.cmd_args
        starttime = time.time()
        child_env = os.environ.copy()
        child_env.update(test.env)
        p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE,
                             env=child_env)
        (stdo, stde) = p.communicate()
        endtime = time.time()
        duration = endtime - starttime
        stdo = stdo.decode()
        stde = stde.decode()
        if p.returncode == 0:
            res = 'OK'
        else:
            res = 'FAIL'
            tests_failed = True
    return TestRun(res, duration, stdo, stde)

def print_stats(numlen, tests, name, result, i, logfile):
    startpad = ' '*(numlen - len('%d' % (i+1)))
    num = '%s%d/%d' % (startpad, i+1, len(tests))
    padding1 = ' '*(40-len(name))
    padding2 = ' '*(5-len(result.res))
    result_str = '%s %s%s%s%s(%5.2f s)' % \
        (num, name, padding1, result.res, padding2, result.duration)
{'fraction_non_alphanumeric': 0.1218568665377176, 'fraction_numerical': 0.008088623175663794, 'mean_word_length': 2.4472727272727273, 'pattern_counts': {'":': 0, '<': 5, '<?xml version=': 0, '>': 5, 'https://': 0, 'lorem ipsum': 0, 'www.': 2, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '7168656', 'n_tokens_mistral': 1808, 'n_tokens_neox': 1779, 'n_words': 551}