diff --git a/src/test/java/de/tla2b/prettyprintb/SpecialDefinitions.java b/src/test/java/de/tla2b/prettyprintb/SpecialDefinitions.java new file mode 100644 index 0000000000000000000000000000000000000000..c3a9d3c901381531c8d24ad257a5efe63cd48163 --- /dev/null +++ b/src/test/java/de/tla2b/prettyprintb/SpecialDefinitions.java @@ -0,0 +1,24 @@ +package de.tla2b.prettyprintb; + +import static de.tla2b.util.TestUtil.compare; + +import org.junit.Test; + +public class SpecialDefinitions { + + + @Test + public void testVisualisationDefinition() throws Exception { + final String module = "-------------- MODULE Testing ----------------\n" + + "EXTENDS Naturals \n" + + "CUSTOM_GRAPH_NODES == 1..10 \n" + + "CUSTOM_GRAPH_EDGES == 1..10 \n" + + "================================="; + + final String expected = "MACHINE Testing\n" + + "DEFINITIONS CUSTOM_GRAPH_NODES == 1 .. 10; \n" + + "CUSTOM_GRAPH_EDGES == 1 .. 10;" + + "END"; + compare(expected, module); + } +}