From 3b41422180259b90ca09c2c2cff324d44b3edc6c Mon Sep 17 00:00:00 2001
From: Mikhail Golubev <mikhail.golubev@jetbrains.com>
Date: Tue, 21 Jun 2016 19:50:06 +0300
Subject: [PATCH] PY-19884 Add a test on type checking of set collection

---
 .../inspections/PyTypeCheckerInspection/SetMethods.py    | 9 +++++++++
 .../python/inspections/PyTypeCheckerInspectionTest.java  | 5 +++++
 2 files changed, 14 insertions(+)
 create mode 100644 python/testData/inspections/PyTypeCheckerInspection/SetMethods.py

diff --git a/python/testData/inspections/PyTypeCheckerInspection/SetMethods.py b/python/testData/inspections/PyTypeCheckerInspection/SetMethods.py
new file mode 100644
index 000000000000..3080f6288cb0
--- /dev/null
+++ b/python/testData/inspections/PyTypeCheckerInspection/SetMethods.py
@@ -0,0 +1,9 @@
+xs = set([1, 2, 3])
+
+'foo' + <warning descr="Expected type 'Union[str, unicode]', got 'int' instead">xs.pop()</warning>
+xs.discard(<weak_warning descr="Expected type 'int' (matched generic type 'TypeVar('T')'), got 'str' instead">'foo'</weak_warning>)
+xs.remove(<weak_warning descr="Expected type 'int' (matched generic type 'TypeVar('T')'), got 'str' instead">'bar'</weak_warning>)
+xs.add(<weak_warning descr="Expected type 'int' (matched generic type 'TypeVar('T')'), got 'object' instead">object()</weak_warning>)
+
+ys = ['green', 'eggs']
+ys.extend(<weak_warning descr="Expected type 'Iterable[str]' (matched generic type 'Iterable[TypeVar('T')]'), got 'Set[int]' instead">xs</weak_warning>)
diff --git a/python/testSrc/com/jetbrains/python/inspections/PyTypeCheckerInspectionTest.java b/python/testSrc/com/jetbrains/python/inspections/PyTypeCheckerInspectionTest.java
index fbf10969bb69..4f4733638284 100644
--- a/python/testSrc/com/jetbrains/python/inspections/PyTypeCheckerInspectionTest.java
+++ b/python/testSrc/com/jetbrains/python/inspections/PyTypeCheckerInspectionTest.java
@@ -300,4 +300,9 @@ public class PyTypeCheckerInspectionTest extends PyTestCase {
   public void testAbsSetAndMutableSet() {
     doTest();
   }
+
+  // PY-19884
+  public void testSetMethods() {
+    doTest();
+  }
 }