Use pytest styled tests

sbellem · sbellem · commit 51cd283724c2 · 2015-11-09T22:17:02.000+01:00
diff --git a/tests.py b/tests.py
@@ -1,12 +1,8 @@
-import unittest
+def test_smile():
+    from awesome import smile
+    assert smile() == ':)'
 
-import awesome
 
-
-class TestMethods(unittest.TestCase):
-    def test_add(self):
-        self.assertEqual(awesome.smile(), ":)")
-
-
-if __name__ == '__main__':
-    unittest.main()
+def test_frown():
+    from awesome import frown
+    assert frown() == ':('
