#!/usr/bin/python

import subprocess
import sys

OUT = "OUT"

Tests = [

     ['1', '9',          {OUT : '3 3'}],
     ['1a', '7',         {OUT : '7'}],
     ['1b', '  +7',      {OUT : '7'}],
     ['2', '4294967291', {OUT : '4294967291'}],
     ['3', '4294967292', {OUT : '2 2 3 3 7 11 31 151 331'}],
     ['4', '4294967293', {OUT : '9241 464773'}],

     ['a', '4294966201', {OUT : '12197 352133'}],
     ['b', '4294966339', {OUT : '13187 325697'}],
     ['c', '4294966631', {OUT : '13729 312839'}],
     ['d', '4294966457', {OUT : '14891 288427'}],
     ['e', '4294966759', {OUT : '21649 198391'}],
     ['f', '4294966573', {OUT : '23071 186163'}],
     ['g', '4294967101', {OUT : '23603 181967'}],
     ['h', '4294966519', {OUT : '34583 124193'}],
     ['i', '4294966561', {OUT : '36067 119083'}],
     ['j', '4294966901', {OUT : '37747 113783'}],
     ['k', '4294966691', {OUT : '39241 109451'}],
     ['l', '4294966969', {OUT : '44201 97169'}],
     ['m', '4294967099', {OUT : '44483 96553'}],
     ['n', '4294966271', {OUT : '44617 96263'}],
     ['o', '4294966789', {OUT : '50411 85199'}],
     ['p', '4294966189', {OUT : '53197 80737'}],
     ['q', '4294967213', {OUT : '57139 75167'}],
     ['s', '4294967071', {OUT : '65521 65551'}],
     ['t', '4294966194', {OUT : '2 3 3 3 3 3 3 3 53 97 191'}],
     ['u', '4294966272', {OUT : '2 2 2 2 2 2 2 2 2 2 3 23 89 683'}],
     ['v', '4294966400', {OUT : '2 2 2 2 2 2 2 5 5 1342177'}],
     ['w', '4294966464', {OUT : '2 2 2 2 2 2 3 3 3 2485513'}],
     ['x', '4294966896', {OUT : '2 2 2 2 3 3 3 11 607 1489'}],
     ['y', '4294966998', {OUT : '2 3 7 3917 26107'}],
     ['bug-2012-a', '465658903', {OUT : '15259 30517'}],
     ['bug-2012-b', '2242724851', {OUT : '33487 66973'}],
     ['bug-2012-c', '6635692801', {OUT : '57601 115201'}],
     ['bug-2012-d', '17709149503', {OUT : '94099 188197'}],
     ['bug-2012-e', '17754345703', {OUT : '94219 188437'}],
     # Infinite loop bugs in v8.20 to 8.26 inclusive
     ['bug-2016-a', '158909489063877810457',
      {OUT : '3401347 3861211 12099721'}],
     ['bug-2016-b', '222087527029934481871',
      {OUT : '15601 26449 111427 4830277'}],
     ['bug-2016-c', '12847291069740315094892340035',
      {OUT : '5 4073 18899 522591721 63874247821'}],
     ['bug-gmp-2_sup_128', '340282366920938463463374607431768211456',
      {OUT : (128 * "2 ")[:-1]}],
     ['bug-gmp-2_sup_256',
      '115792089237316195423570985008687907853'
      + '269984665640564039457584007913129639936',
      {OUT : (256 * "2 ")[:-1]}],
     ['bug-gmp-plus_2_sup_127_plus_1',
      '+170141183460469231731687303715884105729',
      {OUT : '3 56713727820156410577229101238628035243'}],
     ['bug-with-128-bit-uintmax_t',
      '340282366920938463463374607431768211355',
      {OUT : '5 31 2195370109167344925570158757624311041'}],
     ['lt-5000-times-128-bit',
      '1701071552237771378853409662551409288273703',
      {OUT : '4999 340282366920938463463374607431768211297'}],
     ['h-1', '-h 3000', {OUT : '2^3 3 5^3'}],
     ['h-2', '3000 --exponents', {OUT : '2^3 3 5^3'}],
]

ow = sys.stdout.write

rc = 0
n_tests = len(Tests)
ow(f"#(Tests)={n_tests}\n")
for (ti, test_case) in enumerate(Tests):
    ow(f"tested {ti}/{n_tests}\n")
    name = test_case[0]
    s_args = test_case[1]
    out_expected = test_case[2][OUT]
    # ow(f"name={name} s_args={s_args} out_expected={out_expected}\n")
    gnu_cmd = ["factor"] + s_args.split()
    out_gnu_raw = subprocess.check_output(gnu_cmd).decode("utf-8")
    out_gnu = out_gnu_raw.split(': ')[1][:-1]
    if out_gnu != out_expected:
        ow(f"FAIL: name={name} "
           f"(out_gnu={out_gnu} != out_expected={out_expected}\n")
        rc = 1
    uu_cmd = ["./target/debug/coreutils", "factor"] + s_args.split()
    out_uu_raw = subprocess.check_output(uu_cmd).decode("utf-8")
    out_uu = out_uu_raw.split(': ')[1][:-1]
    if out_uu != out_expected:
        ow(f"FAIL: name={name} "
           f"(out_uu={out_uu} != out_expected={out_expected}\n")
        rc = 1

sys.exit(rc)