From a77d713f3fb2c51768bcb036c8b4695bd02103be Mon Sep 17 00:00:00 2001
From: dohan <dohan001@hhu.de>
Date: Mon, 27 Aug 2018 10:22:01 +0200
Subject: [PATCH] test copying of visualisation definitions
 (CUSTOM_GRAPH_EDGES, etc.)

---
 .../prettyprintb/SpecialDefinitions.java      | 24 +++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 src/test/java/de/tla2b/prettyprintb/SpecialDefinitions.java

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 0000000..c3a9d3c
--- /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);
+	}
+}
-- 
GitLab