AbstractOptimizer tests added

opt4j-core/src/test/java/org/opt4j/core/optimizer/AbstractOptimizerTest.java

@@ -0,0 +1,326 @@
+package org.opt4j.core.optimizer;
+
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.Set;
+
+import org.junit.Assert;
+import org.junit.Test;
+import org.opt4j.core.Individual;
+
+public class AbstractOptimizerTest {
+	class MockOptimizer extends AbstractOptimizer {
+		TestMethod t;
+
+		public MockOptimizer(Population population, Archive archive, IndividualCompleter completer, Control control,
+				Iteration iteration) {
+			super(population, archive, completer, control, iteration);
+		}
+
+		public void setOptimizeTest(TestMethod t) {
+			this.t = t;
+		}
+
+		@Override
+		public void optimize() throws StopException, TerminationException {
+			t.test();
+		}
+	}
+
+	private interface TestMethod {
+		public void test() throws TerminationException, StopException;
+	}
+
+	private class MockIndividual extends Individual {
+
+	}
+
+	@Test
+	public void getIteration() {
+		MockOptimizer optimizer = new MockOptimizer(null, null, null, null, new Iteration(2));
+		Assert.assertEquals(0, optimizer.getIteration());
+	}
+
+	@Test
+	public void isRunning() {
+		MockOptimizer optimizer = new MockOptimizer(null, null, null, null, null);
+		Assert.assertFalse(optimizer.isRunning());
+	}
+
+	@Test
+	public void startOptimization() throws TerminationException {
+		final MockOptimizer optimizer = new MockOptimizer(null, null, null, null, null);
+		Assert.assertFalse(optimizer.isRunning());
+		optimizer.setOptimizeTest(new TestMethod() {
+			@Override
+			public void test() throws TerminationException, StopException {
+				Assert.assertTrue(optimizer.isRunning());
+			}
+		});
+		optimizer.startOptimization();
+		Assert.assertFalse(optimizer.isRunning());
+	}
+
+	@Test
+	public void startOptimizationTerminating() throws TerminationException {
+		final Control c = new Control();
+		MockOptimizer optimizer = new MockOptimizer(null, null, null, c, null);
+		Assert.assertFalse(optimizer.isRunning());
+		optimizer.setOptimizeTest(new TestMethod() {
+			@Override
+			public void test() throws TerminationException, StopException {
+				c.doTerminate();
+				c.checkpoint();
+				Assert.fail();
+			}
+		});
+		optimizer.startOptimization();
+		Assert.assertFalse(optimizer.isRunning());
+	}
+
+	boolean started = false;
+
+	@Test
+	public void startOptimizationWithListener() throws TerminationException {
+		final MockOptimizer optimizer = new MockOptimizer(null, null, null, null, null);
+		started = false;
+		final OptimizerStateListener l = new OptimizerStateListener() {
+
+			@Override
+			public void optimizationStarted(Optimizer optimizer) {
+				Assert.assertFalse(started);
+				started = true;
+			}
+
+			@Override
+			public void optimizationStopped(Optimizer optimizer) {
+			}
+
+		};
+		optimizer.addOptimizerStateListener(l);
+
+		optimizer.setOptimizeTest(new TestMethod() {
+			@Override
+			public void test() throws TerminationException, StopException {
+			}
+		});
+		optimizer.startOptimization();
+		Assert.assertTrue(started);
+	}
+
+	boolean archiveHasRun = false;
+	boolean completerHasRun = false;
+
+	@Test
+	public void nextIteration() throws TerminationException {
+		final Control c = new Control();
+		Population p = new Population();
+		final Individual i1 = new MockIndividual();
+		p.add(i1);
+		Archive a = new Archive() {
+
+			@Override
+			public boolean update(Set<? extends Individual> individuals) {
+				archiveHasRun = true;
+				Assert.assertTrue(individuals.contains(i1));
+				return true;
+			}
+		};
+		IndividualCompleter completer = new IndividualCompleter() {
+
+			@Override
+			public void complete(Individual... individuals) throws TerminationException {
+				complete(Arrays.asList(individuals));
+			}
+
+			@Override
+			public void complete(Iterable<? extends Individual> iterable) throws TerminationException {
+				completerHasRun = true;
+				Assert.assertTrue(iterable.iterator().hasNext());
+				Assert.assertSame(i1, iterable.iterator().next());
+			}
+		};
+		final MockOptimizer optimizer = new MockOptimizer(p, a, completer, c, new Iteration(10));
+		Assert.assertFalse(optimizer.isRunning());
+		optimizer.setOptimizeTest(new TestMethod() {
+			@Override
+			public void test() throws TerminationException, StopException {
+				c.doStop();
+				c.checkpoint();
+				optimizer.nextIteration();
+			}
+		});
+		optimizer.startOptimization();
+		Assert.assertFalse(optimizer.isRunning());
+		Assert.assertTrue(archiveHasRun);
+		Assert.assertTrue(completerHasRun);
+	}
+
+	@Test
+	public void nextIterationWithListeners() throws TerminationException {
+		final Control c = new Control();
+		Population p = new Population();
+		final Individual i1 = new MockIndividual();
+		p.add(i1);
+		Archive a = new Archive() {
+			@Override
+			public boolean update(Set<? extends Individual> individuals) {
+				return true;
+			}
+		};
+		IndividualCompleter completer = new IndividualCompleter() {
+
+			@Override
+			public void complete(Individual... individuals) throws TerminationException {
+			}
+
+			@Override
+			public void complete(Iterable<? extends Individual> iterable) throws TerminationException {
+			}
+		};
+		final MockOptimizer optimizer = new MockOptimizer(p, a, completer, c, new Iteration(10));
+
+		started = false;
+		OptimizerIterationListener l1 = new OptimizerIterationListener() {
+			@Override
+			public void iterationComplete(int iteration) {
+				Assert.assertFalse(started);
+				Assert.assertEquals(1, iteration);
+				started = true;
+			}
+		};
+		optimizer.addOptimizerIterationListener(l1);
+
+		Assert.assertFalse(optimizer.isRunning());
+		optimizer.setOptimizeTest(new TestMethod() {
+			@Override
+			public void test() throws TerminationException, StopException {
+				c.doStop();
+				c.checkpoint();
+				optimizer.nextIteration();
+			}
+		});
+		optimizer.startOptimization();
+		Assert.assertTrue(started);
+	}
+
+	@Test
+	public void startOptimizationStopped() throws TerminationException {
+		final Control c = new Control();
+		final MockOptimizer optimizer = new MockOptimizer(null, new Archive() {
+			@Override
+			public boolean update(Set<? extends Individual> individuals) {
+				return false;
+			}
+		}, new IndividualCompleter() {
+			@Override
+			public void complete(Individual... individuals) throws TerminationException {
+			}
+
+			@Override
+			public void complete(Iterable<? extends Individual> iterable) throws TerminationException {
+			}
+		}, c, new Iteration(4));
+		Assert.assertFalse(optimizer.isRunning());
+		optimizer.setOptimizeTest(new TestMethod() {
+			@Override
+			public void test() throws TerminationException, StopException {
+				c.doStop();
+				c.checkpoint();
+				optimizer.nextIteration();
+				Assert.fail();
+			}
+		});
+		optimizer.startOptimization();
+		Assert.assertFalse(optimizer.isRunning());
+	}
+
+	@Test
+	public void addOptimizerIterationListener() {
+		final MockOptimizer optimizer = new MockOptimizer(null, null, null, null, null);
+		OptimizerIterationListener l = new OptimizerIterationListener() {
+			@Override
+			public void iterationComplete(int iteration) {
+			}
+		};
+		optimizer.addOptimizerIterationListener(l);
+
+		Assert.assertEquals(1, optimizer.iterationListeners.size());
+		Assert.assertSame(l, optimizer.iterationListeners.iterator().next());
+	}
+
+	@Test
+	public void addOptimizerStateListener() {
+		final MockOptimizer optimizer = new MockOptimizer(null, null, null, null, null);
+		OptimizerStateListener l = new OptimizerStateListener() {
+			@Override
+			public void optimizationStarted(Optimizer optimizer) {
+			}
+
+			@Override
+			public void optimizationStopped(Optimizer optimizer) {
+			}
+
+		};
+		optimizer.addOptimizerStateListener(l);
+
+		Assert.assertEquals(1, optimizer.stateListeners.size());
+		Assert.assertSame(l, optimizer.stateListeners.iterator().next());
+	}
+
+	@Test
+	public void removeOptimizerIterationListener() {
+		final MockOptimizer optimizer = new MockOptimizer(null, null, null, null, null);
+		OptimizerIterationListener l = new OptimizerIterationListener() {
+			@Override
+			public void iterationComplete(int iteration) {
+			}
+		};
+		optimizer.addOptimizerIterationListener(l);
+		optimizer.removeOptimizerIterationListener(l);
+		Assert.assertTrue(optimizer.iterationListeners.isEmpty());
+	}
+
+	@Test
+	public void removeOptimizerStateListener() {
+		final MockOptimizer optimizer = new MockOptimizer(null, null, null, null, null);
+		OptimizerStateListener l = new OptimizerStateListener() {
+			@Override
+			public void optimizationStarted(Optimizer optimizer) {
+			}
+
+			@Override
+			public void optimizationStopped(Optimizer optimizer) {
+			}
+
+		};
+		optimizer.addOptimizerStateListener(l);
+
+		optimizer.removeOptimizerStateListener(l);
+		Assert.assertTrue(optimizer.stateListeners.isEmpty());
+	}
+
+	@Test
+	public void injectListeners() {
+		final MockOptimizer optimizer = new MockOptimizer(null, null, null, null, null);
+		OptimizerIterationListener l1 = new OptimizerIterationListener() {
+			@Override
+			public void iterationComplete(int iteration) {
+			}
+		};
+		OptimizerStateListener l2 = new OptimizerStateListener() {
+			@Override
+			public void optimizationStarted(Optimizer optimizer) {
+			}
+
+			@Override
+			public void optimizationStopped(Optimizer optimizer) {
+			}
+
+		};
+		optimizer.injectListeners(Collections.singleton(l2), Collections.singleton(l1));
+
+		Assert.assertSame(l1, optimizer.iterationListeners.iterator().next());
+		Assert.assertSame(l2, optimizer.stateListeners.iterator().next());
+	}
+}