X-Git-Url: https://jfr.im/git/yt-dlp.git/blobdiff_plain/10a4858d8f3fd714a2f8c4664eabf06ad7648ea5..90c3721a322756bb7f4ca10ceb73744500bee37e:/test/test_netrc.py

diff --git a/test/test_netrc.py b/test/test_netrc.py
index 566ba37a6..1e0f4ee3b 100644
--- a/test/test_netrc.py
+++ b/test/test_netrc.py
@@ -1,25 +1,27 @@
-# 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'),
-                'Extractor %s supports login, but is missing a _NETRC_MACHINE property' % ie.IE_NAME)
+                ie._NETRC_MACHINE,
+                f'Extractor {ie.IE_NAME} supports login, but is missing a _NETRC_MACHINE property')
 
 
 if __name__ == '__main__':