1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
from periodictable import elements, formula, mix_by_weight, mix_by_volume
from periodictable.core import PeriodicTable
from periodictable import (mass, density, xsf, nsf, formulas,
crystal_structure, covalent_radius)
def test():
# Create private table
private = PeriodicTable("mine")
mass.init(private)
density.init(private)
xsf.init(private)
nsf.init(private)
crystal_structure.init(private)
covalent_radius.init(private)
# Test that it matches public table
assert elements.Cm.mass == private.Cm.mass
assert elements.Cm.density == private.Cm.density
assert elements.Cm.crystal_structure == private.Cm.crystal_structure
assert elements.Cm.covalent_radius == private.Cm.covalent_radius
slde,abse = elements.Cm.xray.sld(energy=8)
sldp,absp = private.Cm.xray.sld(energy=8)
assert slde == sldp
assert abse == absp
# Check that modifications to private table don't change public table.
private.Cm._mass = elements.Cm.mass+1
assert elements.Cm.mass != private.Cm.mass
private.Cm._density = elements.Cm.density+1
assert elements.Cm.density != private.Cm.density
private.Cm.covalent_radius = elements.Cm.covalent_radius+1
assert elements.Cm.covalent_radius != private.Cm.covalent_radius
private.Cm.xray.newfield = 5
assert not hasattr(elements.Cm.xray,'newfield')
# Check that the formula parser can use a private table
privateH2O = formula('H2O', table=private)
publicH2O = formula('H2O', table=elements)
for el in privateH2O.atoms.keys():
assert id(el) == id(private[el.number])
assert privateH2O != publicH2O
assert str(privateH2O) == str(publicH2O)
assert privateH2O == publicH2O.change_table(private)
private.H._mass = 1
assert formula('H2', table=private).mass == 2
|
