Add level switch tests