Add tests for partial infection