Merge pull request #590 from AbdullahM0hamed/split-test

Change split anime test
anime-dl · Dec 16, 2020 · 74e3d37 · 74e3d37
2 parents 93a3bb1 + bc3571b
commit 74e3d37
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_util.py b/tests/test_util.py
@@ -1,15 +1,15 @@
 import pytest
 
 from anime_downloader import util
-
+from unittest.mock import Mock
 
 def test_split_anime():
-    anime_list = list(range(20))
-
-    assert len(util.split_anime(anime_list, '1:10')) == 9
+    anime = Mock()
+    anime._episode_urls = list(enumerate(range(20)))
+    assert len(util.split_anime(anime, '1:10')._episode_urls) == 9
 
 def test_check_in_path_exists():
     assert util.check_in_path('ls')
 
 def test_check_in_path_not_exists():
-    assert not util.check_in_path('someAppI_madeUp') 
+    assert not util.check_in_path('someAppI_madeUp')