test through python assert instead of shell script