diff --git a/tests/test_awslambda/test_policy.py b/tests/test_awslambda/test_policy.py
index d3fe3b0b2..92ef0087b 100644
--- a/tests/test_awslambda/test_policy.py
+++ b/tests/test_awslambda/test_policy.py
@@ -1,7 +1,7 @@
 from __future__ import unicode_literals
 
-import unittest
 import json
+import sure
 
 from moto.awslambda.policy import Policy
 
@@ -12,39 +12,38 @@ class MockLambdaFunction:
         self.policy = None
 
 
-class TestPolicy(unittest.TestCase):
-    def test_policy(self):
-        policy = Policy(MockLambdaFunction("arn"))
-        statement = {
-            "StatementId": "statement0",
-            "Action": "lambda:InvokeFunction",
-            "FunctionName": "function_name",
-            "Principal": "events.amazonaws.com",
-            "SourceArn": "arn:aws:events:us-east-1:111111111111:rule/rule_name",
-            "SourceAccount": "111111111111",
-        }
+def test_policy():
+    policy = Policy(MockLambdaFunction("arn"))
+    statement = {
+        "StatementId": "statement0",
+        "Action": "lambda:InvokeFunction",
+        "FunctionName": "function_name",
+        "Principal": "events.amazonaws.com",
+        "SourceArn": "arn:aws:events:us-east-1:111111111111:rule/rule_name",
+        "SourceAccount": "111111111111",
+    }
 
-        expected = {
-            "Action": "lambda:InvokeFunction",
-            "FunctionName": "function_name",
-            "Principal": {"Service": "events.amazonaws.com"},
-            "Effect": "Allow",
-            "Resource": "arn:$LATEST",
-            "Sid": "statement0",
-            "Condition": {
-                "ArnLike": {
-                    "AWS:SourceArn": "arn:aws:events:us-east-1:111111111111:rule/rule_name",
-                },
-                "StringEquals": {"AWS:SourceAccount": "111111111111"},
+    expected = {
+        "Action": "lambda:InvokeFunction",
+        "FunctionName": "function_name",
+        "Principal": {"Service": "events.amazonaws.com"},
+        "Effect": "Allow",
+        "Resource": "arn:$LATEST",
+        "Sid": "statement0",
+        "Condition": {
+            "ArnLike": {
+                "AWS:SourceArn": "arn:aws:events:us-east-1:111111111111:rule/rule_name",
             },
-        }
+            "StringEquals": {"AWS:SourceAccount": "111111111111"},
+        },
+    }
 
-        policy.add_statement(json.dumps(statement))
-        self.assertDictEqual(expected, policy.statements[0])
+    policy.add_statement(json.dumps(statement))
+    expected.should.be.equal(policy.statements[0])
 
-        sid = statement.get("StatementId", None)
-        if sid == None:
-            raise "TestCase.statement does not contain StatementId"
+    sid = statement.get("StatementId", None)
+    if sid == None:
+        raise "TestCase.statement does not contain StatementId"
 
-        policy.del_statement(sid)
-        self.assertEqual([], policy.statements)
+    policy.del_statement(sid)
+    [].should.be.equal(policy.statements)