hbmartin commited on
Commit
f7b22d6
·
1 Parent(s): 3f026d1
Files changed (1) hide show
  1. tests/test_captions.py +3 -3
tests/test_captions.py CHANGED
@@ -47,7 +47,7 @@ def test_caption_query_get_by_language_code_when_not_exists():
47
  @mock.patch("pytube.captions.Caption.generate_srt_captions")
48
  def test_download(srt):
49
  open_mock = mock_open()
50
- with patch("builtins.open", m):
51
  srt.return_value = ""
52
  caption = Caption(
53
  {"url": "url1", "name": {"simpleText": "name1"}, "languageCode": "en"}
@@ -59,7 +59,7 @@ def test_download(srt):
59
  @mock.patch("pytube.captions.Caption.generate_srt_captions")
60
  def test_download_with_prefix(srt):
61
  open_mock = mock_open()
62
- with patch("builtins.open", m):
63
  srt.return_value = ""
64
  caption = Caption(
65
  {"url": "url1", "name": {"simpleText": "name1"}, "languageCode": "en"}
@@ -71,7 +71,7 @@ def test_download_with_prefix(srt):
71
  @mock.patch("pytube.captions.Caption.xml_captions")
72
  def test_download_xml_and_trim_extension(xml):
73
  open_mock = mock_open()
74
- with patch("builtins.open", m):
75
  xml.return_value = ""
76
  caption = Caption(
77
  {"url": "url1", "name": {"simpleText": "name1"}, "languageCode": "en"}
 
47
  @mock.patch("pytube.captions.Caption.generate_srt_captions")
48
  def test_download(srt):
49
  open_mock = mock_open()
50
+ with patch("builtins.open", open_mock):
51
  srt.return_value = ""
52
  caption = Caption(
53
  {"url": "url1", "name": {"simpleText": "name1"}, "languageCode": "en"}
 
59
  @mock.patch("pytube.captions.Caption.generate_srt_captions")
60
  def test_download_with_prefix(srt):
61
  open_mock = mock_open()
62
+ with patch("builtins.open", open_mock):
63
  srt.return_value = ""
64
  caption = Caption(
65
  {"url": "url1", "name": {"simpleText": "name1"}, "languageCode": "en"}
 
71
  @mock.patch("pytube.captions.Caption.xml_captions")
72
  def test_download_xml_and_trim_extension(xml):
73
  open_mock = mock_open()
74
+ with patch("builtins.open", open_mock):
75
  xml.return_value = ""
76
  caption = Caption(
77
  {"url": "url1", "name": {"simpleText": "name1"}, "languageCode": "en"}