Test list items types

glensc · Mar 8, 2024 · 9d18e21 · 9d18e21
1 parent b9b869f
commit 9d18e21
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_lists.py b/tests/test_lists.py
@@ -1,4 +1,6 @@
 # -*- coding: utf-8 -*-
+from trakt.movies import Movie
+from trakt.tv import TVEpisode, TVSeason, TVShow
 from trakt.users import PublicList
 
 
@@ -13,3 +15,7 @@ def test_public_list():
 
     # Test iter
     assert len(l) == 4
+
+    # enumerate list items
+    instancetypes = (Movie, TVShow, TVSeason, TVEpisode)
+    assert all([isinstance(k.item, instancetypes) for k in l])