More flexible test
Signed-off-by: Sho Nakatani <lay.sakura@gmail.com>