From 7b40bafbd82fef4d6cc62d395040f1770fe202fe Mon Sep 17 00:00:00 2001
From: hagrid67 <jdhwatson@gmail.com>
Date: Tue, 9 Apr 2019 19:13:45 +0100
Subject: [PATCH] adding test_rendertools

---
 tests/test_rendertools.py | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 tests/test_rendertools.py

diff --git a/tests/test_rendertools.py b/tests/test_rendertools.py
new file mode 100644
index 0000000..f331e13
--- /dev/null
+++ b/tests/test_rendertools.py
@@ -0,0 +1,39 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from flatland.core.env import RailEnv
+#from flatland.core.transitions import GridTransitions
+import numpy as np
+import random
+
+from recordtype import recordtype
+
+import numpy as np
+from numpy import array
+import xarray as xr
+import matplotlib.pyplot as plt
+
+from flatland.core.transitions import RailEnvTransitions
+#import flatland.core.env
+from flatland.utils import rail_env_generator
+from flatland.core.env import RailEnv
+import flatland.utils.rendertools as rt
+
+
+
+
+"""Tests for `flatland` package."""
+
+
+
+
+
+def test_render_env():
+    random.seed(100)
+    oRail = rail_env_generator.generate_random_rail(10,10)
+    type(oRail), len(oRail)
+    oEnv = RailEnv(oRail, number_of_agents=2)
+    oEnv.reset()
+    oRT = rt.RenderTool(oEnv)
+    plt.figure(figsize=(10,10))
+    oRT.renderEnv()
-- 
GitLab