Getting ready for v0.3 release.

1 files changed, 10 insertions, 1 deletions

diff --git a/test_archivemail.py b/test_archivemail.py
index 0015541..22ab2f2 100755
--- a/test_archivemail.py
+++ b/test_archivemail.py
@@ -32,6 +32,7 @@ TODO: add tests for:
 
 """
 
+import sys
 import fcntl
 import filecmp
 import os
@@ -41,7 +42,15 @@ import tempfile
 import time
 import unittest
 
-import archivemail
+try:
+    import archivemail
+except ImportError:
+    print "The archivemail script needs to be called 'archivemail.py'"
+    print "and should be in the current directory in order to be imported"
+    print "and tested. Sorry."
+    if os.path.isfile("archivemail"):
+        print "Try renaming it from 'archivemail' to 'archivemail.py'."
+    sys.exit(1)