add multiplication args to test.py
fix redundancy
Merge pull request #1 from maeager/mult

add multiplication args to test.py