[build] Exclude `requests` from `py2exe` (#9982)

[yt-dlp.git] / test / test_netrc.py

diff --git a/test/test_netrc.py b/test/test_netrc.py
index 566ba37a643b94b1156a38e15936748c9b2320b8..dc708d974cf3e743aa45280bb7d5ba944b5b466d 100644 (file)
--- a/test/test_netrc.py
+++ b/test/test_netrc.py
@@ -1,24 +1,26 @@
-# coding: utf-8
-from __future__ import unicode_literals
+#!/usr/bin/env python3
 
+# Allow direct execution
 import os
 import sys
 import unittest
+
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 
-from youtube_dlc.extractor import (
-    gen_extractors,
-)
+from yt_dlp.extractor import gen_extractor_classes
+from yt_dlp.extractor.common import InfoExtractor
+
+NO_LOGIN = InfoExtractor._perform_login
 
 
 class TestNetRc(unittest.TestCase):
     def test_netrc_present(self):
-        for ie in gen_extractors():
-            if not hasattr(ie, '_login'):
+        for ie in gen_extractor_classes():
+            if ie._perform_login is NO_LOGIN:
                 continue
             self.assertTrue(
-                hasattr(ie, '_NETRC_MACHINE'),
+                ie._NETRC_MACHINE,
                 'Extractor %s supports login, but is missing a _NETRC_MACHINE property' % ie.IE_NAME)