Fix datetime mode test

python-rapidjson · Sep 15, 2015 · cd95027 · cd95027
1 parent c1a6730
commit cd95027
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_params.py b/tests/test_params.py
@@ -129,11 +129,14 @@ def test_datetime_mode():
     d = datetime.utcnow()
     dstr = d.isoformat()
 
-    assert rapidjson.dumps(d) == '"%s"' % dstr
+    with pytest.raises(TypeError):
+        rapidjson.dumps(d)
 
     with pytest.raises(TypeError):
         rapidjson.dumps(d, datetime_mode=0)
 
+    assert rapidjson.dumps(d, datetime_mode=1) == '"%s"' % dstr
+
 
 @pytest.mark.unit
 def test_precise_float():