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

5 months ago
"""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)