From e56afb89e409d04ce2113bf5704b33a33b22f62c Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Thu, 19 Sep 2019 08:24:42 +0200
Subject: [PATCH] #178 added type hints
---
tests/test_utils.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/tests/test_utils.py b/tests/test_utils.py
index 0597bbf7..6347bd0f 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -1,15 +1,16 @@
"""Test Utils."""
-from typing import List
+from typing import List, Tuple
from attr import attrs, attrib
+from flatland.core.grid.grid4 import Grid4TransitionsEnum
from flatland.envs.rail_env import RailEnvActions
@attrs
class Replay(object):
- position = attrib()
- direction = attrib()
+ position = attrib(type=Tuple[int, int])
+ direction = attrib(type=Grid4TransitionsEnum)
action = attrib(type=RailEnvActions)
malfunction = attrib(default=0, type=int)
@@ -17,5 +18,5 @@ class Replay(object):
@attrs
class ReplayConfig(object):
replay = attrib(type=List[Replay])
- target = attrib()
+ target = attrib(type=Tuple[int, int])
speed = attrib(type=float)
--
GitLab