add more test code
use Iterable instead of array