From eb92aa94acac666a53adfa209de50adb358b4a94 Mon Sep 17 00:00:00 2001
From: numb3r3 <wangfelix87@gmail.com>
Date: Thu, 19 Aug 2021 13:14:35 +0800
Subject: [PATCH] fix(tests): remove base_image related tests

---
 tests/test_core.py |  6 ++---
 tests/test_deps.py | 56 +++++++++++++++++++++++-----------------------
 2 files changed, 31 insertions(+), 31 deletions(-)

diff --git a/tests/test_core.py b/tests/test_core.py
index 7da464f..e4d55d2 100644
--- a/tests/test_core.py
+++ b/tests/test_core.py
@@ -35,7 +35,7 @@ def test_prelude():
     ]
 
     base_images, tools = core.prelude(imports)
-    assert base_images == set(
-        ['tensorflow/tensorflow:2.5.0', 'pytorch/pytorch:1.6.0-cuda10.2-cudnn7-runtime']
-    )
+    # assert base_images == set(
+    #     ['tensorflow/tensorflow:2.5.0', 'pytorch/pytorch:1.6.0-cuda10.2-cudnn7-runtime']
+    # )
     assert tools == set(['git'])
diff --git a/tests/test_deps.py b/tests/test_deps.py
index f22096c..848f6b8 100644
--- a/tests/test_deps.py
+++ b/tests/test_deps.py
@@ -30,31 +30,31 @@ def test_get_dep_tools():
     assert deps.get_dep_tools(pkg) == ['git']
 
 
-@pytest.mark.parametrize(
-    'package, expect_base_image',
-    [
-        (
-            deps.Package(name='tensorflow', version='2.5.0'),
-            'tensorflow/tensorflow:2.5.0',
-        ),
-        (
-            deps.Package(name='tensorflow-cpu', version='2.1.0'),
-            'tensorflow/tensorflow:2.1.0-py3',
-        ),
-        (
-            deps.Package(name='tensorflow-gpu', version='2.1.0'),
-            'tensorflow/tensorflow:2.1.0-gpu-py3',
-        ),
-        (
-            deps.Package(name='tensorflow-gpu', version='1.12.0'),
-            'tensorflow/tensorflow:1.12.0-gpu-py3',
-        ),
-        (deps.Package(name='pytorch', version='1.8.0+cpu'), 'bitnami/pytorch:1.8.0'),
-        (
-            deps.Package(name='pytorch', version='1.8.0'),
-            'pytorch/pytorch:1.8.0-cuda10.2-cudnn7-runtime',
-        ),
-    ],
-)
-def test_get_base_images(package, expect_base_image):
-    assert deps.get_baseimage(package) == expect_base_image
+# @pytest.mark.parametrize(
+#     'package, expect_base_image',
+#     [
+#         (
+#             deps.Package(name='tensorflow', version='2.5.0'),
+#             'tensorflow/tensorflow:2.5.0',
+#         ),
+#         (
+#             deps.Package(name='tensorflow-cpu', version='2.1.0'),
+#             'tensorflow/tensorflow:2.1.0-py3',
+#         ),
+#         (
+#             deps.Package(name='tensorflow-gpu', version='2.1.0'),
+#             'tensorflow/tensorflow:2.1.0-gpu-py3',
+#         ),
+#         (
+#             deps.Package(name='tensorflow-gpu', version='1.12.0'),
+#             'tensorflow/tensorflow:1.12.0-gpu-py3',
+#         ),
+#         (deps.Package(name='pytorch', version='1.8.0+cpu'), 'bitnami/pytorch:1.8.0'),
+#         (
+#             deps.Package(name='pytorch', version='1.8.0'),
+#             'pytorch/pytorch:1.8.0-cuda10.2-cudnn7-runtime',
+#         ),
+#     ],
+# )
+# def test_get_base_images(package, expect_base_image):
+#     assert deps.get_baseimage(package) == expect_base_image