You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

40 lines
1.3 KiB

"""Tests for functions that inject symbols into the global namespace. """
from sympy.polys.rings import vring
from sympy.polys.fields import vfield
from sympy.polys.domains import QQ
def test_vring():
ns = {'vring':vring, 'QQ':QQ}
exec('R = vring("r", QQ)', ns)
exec('assert r == R.gens[0]', ns)
exec('R = vring("rb rbb rcc rzz _rx", QQ)', ns)
exec('assert rb == R.gens[0]', ns)
exec('assert rbb == R.gens[1]', ns)
exec('assert rcc == R.gens[2]', ns)
exec('assert rzz == R.gens[3]', ns)
exec('assert _rx == R.gens[4]', ns)
exec('R = vring(["rd", "re", "rfg"], QQ)', ns)
exec('assert rd == R.gens[0]', ns)
exec('assert re == R.gens[1]', ns)
exec('assert rfg == R.gens[2]', ns)
def test_vfield():
ns = {'vfield':vfield, 'QQ':QQ}
exec('F = vfield("f", QQ)', ns)
exec('assert f == F.gens[0]', ns)
exec('F = vfield("fb fbb fcc fzz _fx", QQ)', ns)
exec('assert fb == F.gens[0]', ns)
exec('assert fbb == F.gens[1]', ns)
exec('assert fcc == F.gens[2]', ns)
exec('assert fzz == F.gens[3]', ns)
exec('assert _fx == F.gens[4]', ns)
exec('F = vfield(["fd", "fe", "ffg"], QQ)', ns)
exec('assert fd == F.gens[0]', ns)
exec('assert fe == F.gens[1]', ns)
exec('assert ffg == F.gens[2]', ns)