Unit test array merge Python