From 06babebdcfa37e93d675457d6439c8cfbb2e1395 Mon Sep 17 00:00:00 2001
From: Mikhail Golubev <mikhail.golubev@jetbrains.com>
Date: Thu, 25 Dec 2014 17:01:12 +0300
Subject: [PATCH] Always use function parameter as type inference target in
 test data for PyTypeFromUsedAttributesTest

---
 .../jetbrains/python/PyTypeFromUsedAttributesTest.java   | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/python/testSrc/com/jetbrains/python/PyTypeFromUsedAttributesTest.java b/python/testSrc/com/jetbrains/python/PyTypeFromUsedAttributesTest.java
index 2cdea28b20dc..cdfffec888d0 100644
--- a/python/testSrc/com/jetbrains/python/PyTypeFromUsedAttributesTest.java
+++ b/python/testSrc/com/jetbrains/python/PyTypeFromUsedAttributesTest.java
@@ -191,8 +191,8 @@ public class PyTypeFromUsedAttributesTest extends PyTestCase {
                "class MySortable(object):\n" +
                "    def sort(self):\n" +
                "        pass\n" +
-               "x = undefined()\n" +
-               "x.sort()",
+               "def f(x):\n" +
+               "    x.sort()",
                "list | MySortable | OtherClassA | OtherClassB | unknown");
   }
 
@@ -220,9 +220,8 @@ public class PyTypeFromUsedAttributesTest extends PyTestCase {
                "    pass\n" +
                "class C5(C4):\n" +
                "    attr = 'bottom'\n" +
-               "x = undefined()\n" +
-               "x.attr\n" +
-               "x",
+               "def f(x):\n" +
+               "    x.attr\n",
                "C1 | unknown");
   }