optimize imports doesn't remove imports for which inspection is suppressed (PY-5228)

python/src/com/jetbrains/python/inspections/PyUnresolvedReferencesInspection.java

@@ -594,9 +594,13 @@ public class PyUnresolvedReferencesInspection extends PyInspection {
       }
 
       Set<PyImportStatementBase> unusedStatements = new HashSet<PyImportStatementBase>();
+      final PyUnresolvedReferencesInspection suppressableInspection = new PyUnresolvedReferencesInspection();
       for (NameDefiner unusedImport : unusedImports) {
         PyImportStatementBase importStatement = PsiTreeUtil.getParentOfType(unusedImport, PyImportStatementBase.class);
         if (importStatement != null && !unusedStatements.contains(importStatement) && !myUsedImports.contains(importStatement)) {
+          if (suppressableInspection.isSuppressedFor(importStatement)) {
+            continue;
+          }
           // don't remove as unused imports in try/except statements
           if (PsiTreeUtil.getParentOfType(importStatement, PyTryExceptStatement.class) != null) {
             continue;

python/testData/optimizeImports/suppressed.after.py

@@ -0,0 +1,2 @@
+#noinspection PyUnresolvedReferences
+import sys

python/testData/optimizeImports/suppressed.py

@@ -0,0 +1,2 @@
+#noinspection PyUnresolvedReferences
+import sys

python/testSrc/com/jetbrains/python/PyOptimizeImportsTest.java

@@ -35,6 +35,10 @@ public class PyOptimizeImportsTest extends PyTestCase {
   public void testUnresolved() {  // PY-2201
     doTest();
   }
+  
+  public void testSuppressed() {  // PY-5228
+    doTest();
+  } 
 
   private void doTest() {
     myFixture.configureByFile("optimizeImports/" + getTestName(true) + ".py");