From d1d9c01a62d407ba07ade0a200477584ef02f3f1 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner <inform@tiker.net> Date: Sun, 14 Jun 2009 03:39:22 -0400 Subject: [PATCH] Modify tests to make them suitable for py.test. --- test/test_pymbolic.py | 33 ++++++++++++--------------------- 1 file changed, 12 insertions(+), 21 deletions(-) diff --git a/test/test_pymbolic.py b/test/test_pymbolic.py index 5c0aaf3..1638ae1 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 -- GitLab