From 3caebcb5aa3ff6986262df69b1564d348d29aa80 Mon Sep 17 00:00:00 2001
From: Takafumi Arakaki <aka.tkf@gmail.com>
Date: Fri, 15 Mar 2013 13:39:48 +0100
Subject: [PATCH] Add test_basedefinition_type

---
 test/test_api_classes.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 test/test_api_classes.py

diff --git a/test/test_api_classes.py b/test/test_api_classes.py
new file mode 100644
index 00000000..3856d6be
--- /dev/null
+++ b/test/test_api_classes.py
@@ -0,0 +1,23 @@
+import pytest
+
+from jedi import api
+
+
+def make_definitions():
+    return api.defined_names("""
+    import sys
+
+    class C:
+        pass
+
+    x = C()
+
+    def f():
+        pass
+    """)
+
+
+@pytest.mark.parametrize('definition', make_definitions())
+def test_basedefinition_type(definition):
+    assert definition.type in ('module', 'class', 'instance', 'function',
+                               'statement', 'import')