misc test fixes

data-workspaces · May 3, 2020 · bd1a95c · bd1a95c
1 parent 7b7fdc1
commit bd1a95c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/tests/dws-test-environment.yml b/tests/dws-test-environment.yml
@@ -16,3 +16,6 @@ dependencies:
     - mypy
     - pyflakes
     - black
+    - click
+    - requests
+    - dws-repo2docker
diff --git a/tests/test_tensorflow.py b/tests/test_tensorflow.py
@@ -105,7 +105,7 @@ def test_wrapper_for_dataset(self):
         import pandas as pd
         import numpy as np
         self._setup_initial_repo(git_resources='results', api_resources='titanic-data')
-        TRAIN_DATA_URL = "https://storage.googleapis.com/tf-daxtasets/titanic/train.csv"
+        TRAIN_DATA_URL = "https://storage.googleapis.com/tf-datasets/titanic/train.csv"
         TEST_DATA_URL = "https://storage.googleapis.com/tf-datasets/titanic/eval.csv"
         train_file_path = tf.keras.utils.get_file("train.csv", TRAIN_DATA_URL)
         test_file_path = tf.keras.utils.get_file("eval.csv", TEST_DATA_URL)

diff --git a/tests/utils_for_tests.py b/tests/utils_for_tests.py
@@ -8,6 +8,7 @@
 from os.path import join, abspath, expanduser, exists
 import os
 import unittest
+import sys
 
 TEMPDIR=abspath(expanduser(__file__)).replace('.py', '_data')
 WS_DIR=join(TEMPDIR,'workspace')