def diffbehavior(args: argparse.Namespace, options: AnalysisOptions,
stdout: TextIO, stderr: TextIO) -> int:
def checked_load(qualname: str) -> Optional[FunctionInfo]:
try:
objs = list(load_files_or_qualnames([qualname]))
except Exception as exc:
print(f'Unable to load "{qualname}": {exc}', file=stderr)
return None
obj = objs[0]
if not isinstance(obj, FunctionInfo):
print(f'"{qualname}" does not target a function.', file=stderr)
return None
return obj
(fn_name1, fn_name2) = (args.fn1, args.fn2)
fn1 = checked_load(fn_name1)
fn2 = checked_load(fn_name2)
if fn1 is None or fn2 is None:
return 2
options.stats = collections.Counter()
diffs = diff_behavior(fn1, fn2, options)
debug("stats", options.stats)
if isinstance(diffs, str):
print(diffs, file=stderr)
return 2
elif len(diffs) == 0:
num_paths = options.stats["num_paths"]
exhausted = options.stats["exhaustion"] > 0
stdout.write(
f"No differences found. (attempted {num_paths} iterations)\n")
if exhausted:
stdout.write(
"All paths exhausted, functions are likely the same!\n")
else:
stdout.write(
"Consider trying longer with: --per_condition_timeout=<seconds>\n"
)
return 0
else:
width = max(len(fn_name1), len(fn_name2)) + 2
for diff in diffs:
inputs = ", ".join(f"{k}={v}" for k, v in diff.args.items())
stdout.write(f"Given: ({inputs}),\n")
result1, result2 = diff.result1, diff.result2
differing_args = result1.get_differing_arg_mutations(result2)
stdout.write(
f"{fn_name1.rjust(width)} : {result1.describe(differing_args)}\n"
)
stdout.write(
f"{fn_name2.rjust(width)} : {result2.describe(differing_args)}\n"
)
return 1
def cover(args: argparse.Namespace, options: AnalysisOptions, stdout: TextIO,
stderr: TextIO) -> int:
ctxfn = checked_load(args.fn, stderr)
if ctxfn is None:
return 2
options.stats = collections.Counter()
paths = path_cover(ctxfn, options, args.coverage_type)
fn, _ = ctxfn.callable()
example_output_format = args.example_output_format
if example_output_format == ExampleOutputFormat.ARGUMENT_DICTIONARY:
return output_argument_dictionary_paths(fn, paths, stdout, stderr)
elif example_output_format == ExampleOutputFormat.EVAL_EXPRESSION:
return output_eval_exression_paths(fn, paths, stdout, stderr)
if example_output_format == ExampleOutputFormat.PYTEST:
return output_pytest_paths(fn, paths, stdout, stderr)
assert False
def diffbehavior(args: argparse.Namespace, options: AnalysisOptions,
stdout: TextIO, stderr: TextIO) -> int:
(fn_name1, fn_name2) = (args.fn1, args.fn2)
fn1 = checked_load(fn_name1, stderr)
fn2 = checked_load(fn_name2, stderr)
if fn1 is None or fn2 is None:
return 2
options.stats = collections.Counter()
diffs = diff_behavior(fn1, fn2, options)
debug("stats", options.stats)
if isinstance(diffs, str):
print(diffs, file=stderr)
return 2
elif len(diffs) == 0:
num_paths = options.stats["num_paths"]
exhausted = options.stats["exhaustion"] > 0
stdout.write(
f"No differences found. (attempted {num_paths} iterations)\n")
if exhausted:
stdout.write(
"All paths exhausted, functions are likely the same!\n")
else:
stdout.write(
"Consider trying longer with: --per_condition_timeout=<seconds>\n"
)
return 0
else:
width = max(len(fn_name1), len(fn_name2)) + 2
for diff in diffs:
inputs = ", ".join(f"{k}={v}" for k, v in diff.args.items())
stdout.write(f"Given: ({inputs}),\n")
result1, result2 = diff.result1, diff.result2
differing_args = result1.get_differing_arg_mutations(result2)
stdout.write(
f"{fn_name1.rjust(width)} : {result1.describe(differing_args)}\n"
)
stdout.write(
f"{fn_name2.rjust(width)} : {result2.describe(differing_args)}\n"
)
return 1