diff --git a/test/test_pymbolic.py b/test/test_pymbolic.py
index 5c0aaf33407d21f100785290a261075ca608283c..1638ae1e6bbf1b07feae59817fc9b494a1e604ce 100644
--- a/test/test_pymbolic.py
+++ b/test/test_pymbolic.py
@@ -1,21 +1,12 @@
-import unittest
-
-
-
-
-class TestPymbolic(unittest.TestCase):
-    def test_expand(self):
-        from pymbolic import var, expand
-
-        x = var("x")
-        u = (x+1)**5
-        expand(u)
-
-    def test_substitute(self):
-        from pymbolic import parse, substitute, evaluate
-        u = parse("5+x.min**2")
-        xmin = parse("x.min")
-        assert evaluate(substitute(u, {xmin:25})) == 630
-            
-if __name__ == '__main__':
-    unittest.main()
+def test_expand():
+    from pymbolic import var, expand
+
+    x = var("x")
+    u = (x+1)**5
+    expand(u)
+
+def test_substitute():
+    from pymbolic import parse, substitute, evaluate
+    u = parse("5+x.min**2")
+    xmin = parse("x.min")
+    assert evaluate(substitute(u, {xmin:25})) == 630