only delete .rst and .png in the tests directory

ivanov · ivanov · commit 6d0b2c42a5b6 · 2012-03-13T17:23:29.000-07:00
diff --git a/tests/test_simple.py b/tests/test_simple.py
@@ -6,8 +6,8 @@
 
 def clean_dir():
     "Remove .rst files created during conversion"
-    map(os.remove, glob.glob("*.rst"))
-    map(os.remove, glob.glob("*.png"))
+    map(os.remove, glob.glob("./tests/*.rst"))
+    map(os.remove, glob.glob("./tests/*.png"))
 
 
 @nt.with_setup(clean_dir, clean_dir)
