Add switch tests