#!/usr/bin/python

import shtest, sys, common
from common import *

def mul(p, q, types=[]):
    if is_array(p) and is_array(q):
        result = [a * b for (a,b) in upzip(p, q)]
    elif is_array(q):
        result = [p * b for b in q]
    else:
        result = [a * q for a in p]
    return shtest.make_test(result, [p, q], types)

def insert_into(test, modifying=False):
    test.add_test(mul((0.0, 1.0, 2.0), (3.0, 4.0, 5.0)))
    test.add_test(mul((1.0, 2.0, 3.0), (7.0,)))
    test.add_test(mul((1.0,), (7.0,)))
    test.add_test(mul((4.0, 5.0, 6.0), (-4.0, -5.0, -6.0)))

    # 1 * N
    if not modifying:
        test.add_test(mul((1.0,), (3.0, 4.0, 5.0)))
        test.add_test(mul(0.5, (10.0, 56.0, -0.5)))
        test.add_test(mul(2.3, (10.0, 56.0, -0.5)))
        test.add_test(mul(0.0, (10.0, 56.0, -0.5)))

    # N * 1
    test.add_test(mul((10.0, 56.0, -0.5), -0.1))
    test.add_test(mul((10.0, 56.0, -0.5), -1.4))

# Test regular multiplication in stream programs
test = shtest.StreamTest('mul', 2)
test.add_call(shtest.Call(shtest.Call.infix, '*', 2))
insert_into(test)
test.output_header(sys.stdout)
test.output(sys.stdout, False)

# Test modifying multiplication in stream programs
test = shtest.StreamTest('mmul', 2)
test.add_call(shtest.Call(shtest.Call.infix, '*=', 2))
insert_into(test, True)
test.output(sys.stdout, False)

# Test regular multiplication in immediate mode
test = shtest.ImmediateTest('mul_im', 2)
test.add_call(shtest.Call(shtest.Call.infix, '*', 2))
insert_into(test)
test.output(sys.stdout, False)

# Test modifying multiplication in immediate mode
test = shtest.ImmediateTest('mmul_im', 2)
test.add_call(shtest.Call(shtest.Call.infix, '*=', 2))
insert_into(test, True)
test.output(sys.stdout, False)

test.output_footer(sys.stdout)

@@ -569,24 +569,297 @@
 0, -6.0)))%0A%0A
+    test.add_test(mul((0.0, 1.0, 2.0), (3.0, 4.0, 5.0), %5B'i', 'i', 'i'%5D))%0A    test.add_test(mul((1.0, 2.0, 3.0), (7.0,), %5B'i', 'i', 'i'%5D))%0A    test.add_test(mul((1.0,), (7.0,), %5B'i', 'i', 'i'%5D))%0A    test.add_test(mul((4.0, 5.0, 6.0), (-4.0, -5.0, -6.0), %5B'i', 'i', 'i'%5D))%0A%0A
     # 1 * N%0A

Add some integer test cases. Have fun Barry!