Test median

src/main/java/Median.java

@@ -1,10 +1,33 @@
 /*
    Написать метод, который принимает на вход массив целых чисел и возвращает медиану (серединное значение)
+
+   Медианой называется число, стоящее в упорядоченном ряде чисел посередине.
+   Например, в ряду: 1, 2, 3, 7, 9 — тройка является медианой. Если количество чисел чётное,
+   то за медиану принимают среднее двух стоящих в центре чисел.
+   Либо можно выбирать элемент под номером k/2, если k чётное и (k+1)/2, если нечетное.
 */
 
+import java.util.Arrays;
+
 public class Median {
     public int median(int[] numbers){
+        if (numbers == null) {
+            return Integer.MAX_VALUE;
+        }
+        if (numbers.length == 0){
+            return Integer.MIN_VALUE;
+        } else {
+            Arrays.sort(numbers);
+            int median;
+            if (numbers.length % 2 == 0) {
+                int middleIndex = numbers.length / 2;
+                median = (numbers[middleIndex] + numbers[middleIndex - 1]) /2;
+            } else {
+                int middleIndex = numbers.length / 2;
+                median = numbers[middleIndex];
+            }
+            return median;
+        }
 
-        return 1;
     }
 }

src/test/java/DataNumbersProvider.java

@@ -0,0 +1,43 @@
+import org.testng.annotations.DataProvider;
+
+public class DataNumbersProvider {
+
+    @DataProvider(name = "validNumberArrayProvider")
+    public Object[][] validNumberArrayProvider() {
+        return new Object[][]{
+                {new int[]{1, 2, 3, 4}, 2.5},
+                {new int[]{-1, -2, -3, -4}, -2.5},
+                {new int[]{1, 3, 5}, 3.0},
+                {new int[]{5, 3, 2}, 3.0},
+                {new int[]{1, 2, 3, 4, 5, 6}, 3.5},
+                {new int[]{6, 5, 4, 3, 2, 1}, 3.5},
+                {new int[]{3, 1, 4, 1, 5, 9}, 3.5},
+                {new int[]{3, 1, 4, 1, 5}, 3.0},
+                {new int[]{-1, -2, -3, -4, -5}, -3.0},
+                {new int[]{2147483647, 2147483647, 2147483647, 2147483647}, 2147483647},
+                {new int[]{2147483647, 2147483647, 2147483647}, 2147483647},
+                {new int[]{2147483647}, 2147483647},
+                {new int[]{20, 18, 16, 14, 12, 10, 8, 6, 4, 2}, 11.0},
+                {new int[]{1000000, 2000000, 3000001, 4000000, 5000000}, 3000001.0},
+        };
+    }
+
+    @DataProvider(name = "invalidNumberArrayProvider")
+    public Object[][] invalidNumberArrayProvider() {
+        return new Object[][]{
+                {null, 0x80000000},
+                {new int[]{}, 0x80000000},
+        };
+    }
+
+    @DataProvider(name = "singleElementArrayProvider")
+    public Object[][] singleElementArrayProvider() {
+        return new Object[][]{
+                {new int[]{0}, 0.0},
+                {new int[]{1}, 1.0},
+                {new int[]{2}, 2.0},
+                {new int[]{5}, 5.0},
+                {new int[]{-5}, -5.0},
+        };
+    }
+}

src/test/java/MedianTest.java

@@ -1,7 +1,43 @@
+import org.testng.annotations.BeforeClass;
 import org.testng.annotations.Test;
 
+import static org.testng.AssertJUnit.assertEquals;
+
 public class MedianTest {
+
+    private Median median;
+    @BeforeClass
+    public void beforeClass() {
+        median = new Median();
+    }
+
     @Test
     public void testMedianHN() {
     }
+
+    @Test(dataProvider = "validNumberArrayProvider",
+            dataProviderClass = DataNumbersProvider.class)
+    public void testFindMedianValidNumberArrayProvider(int[] input, double expected) {
+        double actual = median.median(input);
+        assertEquals(expected, actual);
+    }
+    @Test(dataProvider = "validNumberArrayProvider",
+            dataProviderClass = DataNumbersProvider.class)
+    public void testFindMedianValidNumberArrayProviderQuickSelect(int[] input, double expected) {
+        double actual = median.median(input);
+        assertEquals(expected, actual);
+    }
+    @Test(dataProvider = "invalidNumberArrayProvider",
+            dataProviderClass = DataNumbersProvider.class)
+    public void testCalculateMedianWithInvalidInput(int[] numbers, double expected) {
+        double actual = median.median(numbers);
+        assertEquals(expected, actual);
+    }
+
+    @Test(dataProvider = "singleElementArrayProvider",
+            dataProviderClass = DataNumbersProvider.class)
+    public void testFindMedianValidNumberArrayProviderWithSingleElementArrayProvider(int[] input, double expected) {
+        double actual = median.median(input);
+        assertEquals(actual, expected);
+    }
 }