package g0901_1000.s0924_minimize_malware_spread;

import static org.hamcrest.CoreMatchers.equalTo;
import static org.hamcrest.MatcherAssert.assertThat;

import org.junit.jupiter.api.Test;

class SolutionTest {
    @Test
    void minMalwareSpread() {
        assertThat(
                new Solution()
                        .minMalwareSpread(
                                new int[][] {{1, 1, 0}, {1, 1, 0}, {0, 0, 1}}, new int[] {0, 1}),
                equalTo(0));
    }

    @Test
    void minMalwareSpread2() {
        assertThat(
                new Solution()
                        .minMalwareSpread(
                                new int[][] {{1, 0, 0}, {0, 1, 0}, {0, 0, 1}}, new int[] {0, 2}),
                equalTo(0));
    }

    @Test
    void minMalwareSpread3() {
        assertThat(
                new Solution()
                        .minMalwareSpread(
                                new int[][] {{1, 1, 1}, {1, 1, 1}, {1, 1, 1}}, new int[] {1, 2}),
                equalTo(1));
    }
}
