diff --git a/src/main/java/edu/kit/typicalc/view/main/TypeAssumptionsArea.java b/src/main/java/edu/kit/typicalc/view/main/TypeAssumptionsArea.java index 7f8fca3..c86b616 100644 --- a/src/main/java/edu/kit/typicalc/view/main/TypeAssumptionsArea.java +++ b/src/main/java/edu/kit/typicalc/view/main/TypeAssumptionsArea.java @@ -69,16 +69,8 @@ public class TypeAssumptionsArea extends Dialog implements LocaleChangeObserver assumptionContainer = new VerticalLayout(); assumptionContainer.setId(ASS_CONTAINER_ID); - - for (Map.Entry param : types.entrySet()) { - TypeAssumptionField assumption = new TypeAssumptionField(value -> { - assumptionContainer.remove(value); - fields.remove(value); - }, param.getKey(), param.getValue()); - assumptionContainer.add(assumption); - fields.add(assumption); - } - + + initializeWithAssumptions(types); layout.add(heading, buttons, assumptionContainer); add(layout); setCloseOnOutsideClick(false); @@ -90,6 +82,17 @@ public class TypeAssumptionsArea extends Dialog implements LocaleChangeObserver protected TypeAssumptionsArea() { this(new HashMap<>()); } + + private void initializeWithAssumptions(Map types) { + for (Map.Entry param : types.entrySet()) { + TypeAssumptionField assumption = new TypeAssumptionField(value -> { + assumptionContainer.remove(value); + fields.remove(value); + }, param.getKey(), param.getValue()); + assumptionContainer.add(assumption); + fields.add(assumption); + } + } private void onAddAssumptionClicked() { TypeAssumptionField assumption = new TypeAssumptionField(value -> {