From 9ea8f0d570c802671a9cb33f580f91ca71245522 Mon Sep 17 00:00:00 2001
From: wassname <wassname@users.noreply.github.com>
Date: Tue, 14 Nov 2017 12:08:10 +0800
Subject: [PATCH] test scaling

---
 test/test_env.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/test/test_env.py b/test/test_env.py
index 2b65c54..504cdb1 100644
--- a/test/test_env.py
+++ b/test/test_env.py
@@ -34,6 +34,9 @@ def test_src_outputs(spec_id):
         raise AssertionError(
             "These should not be equal, or we are giving future prices to the model")
 
+    # also make sure that the last price X, is 1 because we scaled it by itself as in eq 18
+    assert (X0[:, -1, 0] == 1).all()
+
 
 @pytest.mark.parametrize("spec_id", env_specs)
 def test_gym_env(spec_id):