tests for FlowDynamics

src/main/java/fr/inrae/agriterix/simulator/Simulator.java

@@ -122,7 +122,7 @@ public class Simulator {
         for (Class type : new Class[]{Product.class, Patch.class, GeoArea.class, Flow.class, Livestock.class,
                 Ratio.class, Tag.class, Farmer.class, Farm.class, Unit.class, CulturalPractice.class, Parameters.class,
                 Simulator.class, Product.class, MarketDynamics.class, ProductionDynamics.class,
-                PopulationDynamics.class, FlowDynamics.class, Dynamics.class}) {
+                PopulationDynamicsByAgeClass.class, FlowDynamics.class, Dynamics.class}) {
             xstream.alias(namespace + type.getSimpleName(), type);
         }
         // useful w/ xpath

src/main/java/fr/inrae/agriterix/simulator/dynamics/PopulationDynamicsByAgeClass.java

@@ -18,7 +18,8 @@ public class PopulationDynamicsByAgeClass extends Dynamics {
     public PopulationDynamicsByAgeClass() {
     }
 
-    public PopulationDynamicsByAgeClass(double amplitude, double drop_x, double mu, double v, double r_threshold, double r_factor, TransmissionStrategy transmission) {
+    public PopulationDynamicsByAgeClass(double amplitude, double drop_x, double mu, double v, double r_threshold,
+                                        double r_factor, TransmissionStrategy transmission) {
         this.amplitude = amplitude;
         this.drop_x = drop_x;
         this.mu = mu;

src/test/java/fr/inrae/agriterix/simulator/DynamicsTest.java

 package fr.inrae.agriterix.simulator;
 
-import com.opencsv.CSVReader;
-import com.opencsv.exceptions.CsvException;
+import fr.inrae.agriterix.simulator.dynamics.FlowDynamics;
 import fr.inrae.agriterix.simulator.dynamics.MarketDynamics;
-import fr.inrae.agriterix.simulator.dynamics.PopulationDynamics;
 import org.junit.*;
 
 import static org.junit.Assert.*;
 
-import java.io.FileNotFoundException;
-import java.io.FileReader;
-import java.io.IOException;
 import java.util.*;
-import java.util.concurrent.ConcurrentHashMap;
 
 public class DynamicsTest {
 
@@ -37,6 +31,69 @@ public class DynamicsTest {
     }
 
 
+    @Test
+    public void testEditInputs() {
+        List<Flow.Input> inputs = new ArrayList<>();
+        inputs.add(new Flow.Input(new Product("Raisin", Unit.Kg), 100f));
+        List<Product> outputs = new ArrayList<>();
+        outputs.add(new Product("Vin", Unit.L));
+        Map<Product, Float> required = new HashMap<>();
+        required.put(new Product("Raisin", Unit.Kg), 1f);
+        Map<Product, Float> produced = new HashMap<>();
+        produced.put(new Product("Vin", Unit.L), 1f);
+        Stocks stocks = new Stocks();
+        Ratio ratio = new Ratio(required, produced);
+        Flow flow = new Flow(inputs, stocks, outputs, ratio);
+        FlowDynamics flowDynamics = new FlowDynamics();
+        flowDynamics.initLogger();
+        Map<String, Float> edits = new HashMap<>();
+        edits.put("Raisin",  0.5f);
+        flowDynamics.editInputs(0, flow, edits);
+        assertEquals(Optional.of(flow.getInputs().get(0).getMaximum()), Optional.of(50f));
+    }
+
+    @Test
+    public void testEditProduced() {
+        List<Flow.Input> inputs = new ArrayList<>();
+        inputs.add(new Flow.Input(new Product("Raisin", Unit.Kg), 100f));
+        List<Product> outputs = new ArrayList<>();
+        outputs.add(new Product("Vin", Unit.L));
+        Map<Product, Float> required = new HashMap<>();
+        required.put(new Product("Raisin", Unit.Kg), 1f);
+        Map<Product, Float> produced = new HashMap<>();
+        produced.put(new Product("Vin", Unit.L), 1f);
+        Stocks stocks = new Stocks();
+        Ratio ratio = new Ratio(required, produced);
+        Flow flow = new Flow(inputs, stocks, outputs, ratio);
+        FlowDynamics flowDynamics = new FlowDynamics();
+        flowDynamics.initLogger();
+        Map<String, Float> edits = new HashMap<>();
+        edits.put("Vin",  0.5f);
+        flowDynamics.editProduced(0, flow, edits);
+        assertEquals(Optional.of(flow.getRatio().getProduced().get(new Product("Vin", Unit.L))), Optional.of(.5f));
+    }
+
+    @Test
+    public void testEditRequired() {
+        List<Flow.Input> inputs = new ArrayList<>();
+        inputs.add(new Flow.Input(new Product("Raisin", Unit.Kg), 100f));
+        List<Product> outputs = new ArrayList<>();
+        outputs.add(new Product("Vin", Unit.L));
+        Map<Product, Float> required = new HashMap<>();
+        required.put(new Product("Raisin", Unit.Kg), 1f);
+        Map<Product, Float> produced = new HashMap<>();
+        produced.put(new Product("Vin", Unit.L), 1f);
+        Stocks stocks = new Stocks();
+        Ratio ratio = new Ratio(required, produced);
+        Flow flow = new Flow(inputs, stocks, outputs, ratio);
+        FlowDynamics flowDynamics = new FlowDynamics();
+        flowDynamics.initLogger();
+        Map<String, Float> edits = new HashMap<>();
+        edits.put("Raisin",  0.5f);
+        flowDynamics.editRequirements(0, flow, edits);
+        assertEquals(Optional.of(flow.getRatio().getRequired().get(new Product("Raisin", Unit.Kg))), Optional.of(.5f));
+    }
+
     @Test
     public void testAllocate() {
         // debug : String bold = "\033[0;1m";