Update show/season watched/unwatched tests

Author: JonnyWong16

tests/test_video.py

@@ -611,17 +611,21 @@ def test_video_Show_history(show):
 
 def test_video_Show_watched(tvshows):
     show = tvshows.get("The 100")
-    show.episodes()[0].markWatched()
+    episode = show.episodes()[0]
+    episode.markWatched()
     watched = show.watched()
     assert len(watched) == 1 and watched[0].title == "Pilot"
+    episode.markUnwatched()
 
 
 def test_video_Show_unwatched(tvshows):
     show = tvshows.get("The 100")
     episodes = show.episodes()
-    episodes[0].markWatched()
+    episode = episodes[0]
+    episode.markWatched()
     unwatched = show.unwatched()
     assert len(unwatched) == len(episodes) - 1
+    episode.markUnwatched()
 
 
 def test_video_Show_settings(show):
@@ -885,6 +889,25 @@ def test_video_Season_history(show):
     season.markUnwatched()
 
 
+def test_video_Season_watched(tvshows):
+    season = tvshows.get("The 100").season(1)
+    episode = season.episode(1)
+    episode.markWatched()
+    watched = season.watched()
+    assert len(watched) == 1 and watched[0].title == "Pilot"
+    episode.markUnwatched()
+
+
+def test_video_Season_unwatched(tvshows):
+    season = tvshows.get("The 100").season(1)
+    episodes = season.episodes()
+    episode = episodes[0]
+    episode.markWatched()
+    unwatched = season.unwatched()
+    assert len(unwatched) == len(episodes) - 1
+    episode.markUnwatched()
+
+
 def test_video_Season_attrs(show):
     season = show.season("Season 1")
     assert utils.is_datetime(season.addedAt)