MeansTest.java

packagecom.thealgorithms.maths;

importstaticorg.junit.jupiter.api.Assertions.assertEquals;
importstaticorg.junit.jupiter.api.Assertions.assertThrows;

importjava.util.ArrayList;
importjava.util.LinkedHashSet;
importjava.util.LinkedList;
importjava.util.List;
importjava.util.Set;
importjava.util.Vector;
importorg.assertj.core.util.Lists;
importorg.assertj.core.util.Sets;
importorg.junit.jupiter.api.Test;

classMeansTest {

@Test
voidarithmeticMeanZeroNumbers() throwsIllegalArgumentException {
List<Double> numbers = newArrayList<>();
assertThrows(IllegalArgumentException.class, () -> Means.arithmetic(numbers));
 }

@Test
voidgeometricMeanZeroNumbers() throwsIllegalArgumentException {
List<Double> numbers = newArrayList<>();
assertThrows(IllegalArgumentException.class, () -> Means.geometric(numbers));
 }

@Test
voidharmonicMeanZeroNumbers() throwsIllegalArgumentException {
List<Double> numbers = newArrayList<>();
assertThrows(IllegalArgumentException.class, () -> Means.harmonic(numbers));
 }

@Test
voidarithmeticMeanSingleNumber() {
List<Double> numbers = Lists.newArrayList(2.5);
assertEquals(2.5, Means.arithmetic(numbers));
 }

@Test
voidgeometricMeanSingleNumber() {
Set<Double> numbers = Sets.newHashSet(Lists.newArrayList(2.5));
assertEquals(2.5, Means.geometric(numbers));
 }

@Test
voidharmonicMeanSingleNumber() {
LinkedHashSet<Double> numbers = Sets.newLinkedHashSet(2.5);
assertEquals(2.5, Means.harmonic(numbers));
 }

@Test
voidarithmeticMeanMultipleNumbers() {
Set<Double> numbers = Sets.newTreeSet(1d, 2.5, 83.3, 25.9999, 46.0001, 74.7, 74.5);
assertEquals(44, Means.arithmetic(numbers));
 }

@Test
voidgeometricMeanMultipleNumbers() {
LinkedList<Double> numbers = newLinkedList<>(Lists.newArrayList(1d, 2d, 3d, 4d, 5d, 6d, 1.25));
assertEquals(2.6426195539300585, Means.geometric(numbers));
 }

@Test
voidharmonicMeanMultipleNumbers() {
Vector<Double> numbers = newVector<>(Lists.newArrayList(1d, 2.5, 83.3, 25.9999, 46.0001, 74.7, 74.5));
assertEquals(4.6697322801074135, Means.harmonic(numbers));
 }
}