From 1e4fe16b72959797e08e206d29125ef81ef3d758 Mon Sep 17 00:00:00 2001
From: RussellBallestrini <devnull@localhost>
Date: Sat, 12 Jan 2013 14:27:23 -0500
Subject: [PATCH] Fixed test to use types

---
 test_ago.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/test_ago.py b/test_ago.py
index 422b6c5..09842d2 100644
--- a/test_ago.py
+++ b/test_ago.py
@@ -1,10 +1,12 @@
+from datetime import datetime
+from datetime import timedelta
+
+from types import *
+
 from ago import human
 from ago import delta2human
 from ago import delta2dict
 
-from datetime import datetime
-from datetime import timedelta
-
 PRESENT = datetime.now()
 PAST    = PRESENT - timedelta( 492, 58711, 45 ) # days, secs, ms
 FUTURE  = PRESENT + timedelta( 2, 12447, 963 ) # days, secs, ms
@@ -13,13 +15,13 @@ PAST_DELTA   = PRESENT - PAST
 FUTURE_DELTA = PRESENT - FUTURE
 
 def test_human_is_string():
-    assert 'str' in str(type(human( PAST )))
+    assert type(human( PAST )) is StringType
 
 def test_delta2human_is_string():
-    assert 'str' in str(type(delta2human( PAST_DELTA )))
+    assert type(delta2human( PAST_DELTA )) is StringType
 
 def test_delta2dict_is_dict():
-    assert 'dict' in str(type(delta2dict( PAST_DELTA )))
+    assert type(delta2dict( PAST_DELTA )) is DictType
 
 def test_ago_in_past_human():
     assert 'ago' in human( PAST )