From 8ce6a13292b75c9be222d3a25633f6d000f375aa Mon Sep 17 00:00:00 2001
From: wuhanstudio <wuhanstudio@qq.com>
Date: Tue, 26 Oct 2021 16:47:50 +0100
Subject: [PATCH] [ci] use cat as default input

---
 tests/test_simba.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_simba.py b/tests/test_simba.py
index 1fcab61..3b3e447 100644
--- a/tests/test_simba.py
+++ b/tests/test_simba.py
@@ -13,7 +13,7 @@ class TestSimBA(unittest.TestCase):
 
     def test_simba(self):
         # Load Image [0.0, 1.0]
-        x = np.asarray(Image.open("tests/dog.jpg").resize((32, 32))) / 255.0
+        x = np.asarray(Image.open("tests/cat.jpg").resize((32, 32))) / 255.0
 
         # Initialize API Model
         model = VGG16Cifar10("https://api.wuhanstudio.uk" + "/vgg16_cifar10")
@@ -21,7 +21,7 @@ def test_simba(self):
         # Get Preditction
         y_pred = model.predict(np.array([x]))[0]
 
-        assert (np.argmax(y_pred) == 5)
+        assert (np.argmax(y_pred) == 3)
 
         # SimBA Attack
         simba = SimBA(model)