def test___getstate__(self):
"""check OverlapCalculator.__getstate__()
"""
olc = self.olc
self.assertIs(None, olc.__getstate__()[-1])
tb = CovalentRadiiTable()
olc.atomradiitable = tb
self.assertIs(tb, olc.__getstate__()[-1])
olc.atomradiitable = "constant"
self.assertIs(None, olc.__getstate__()[-1])
return
def test_pickling_artb(self):
'''check pickling and unpickling of OverlapCalculator.atomradiitable.
'''
olc = self.olc
olc.atomradiitable.setDefault(1.3)
spkl = pickle.dumps(olc)
olc1 = pickle.loads(spkl)
self.assertFalse(olc is olc1)
self.assertEqual(1.3, olc1.atomradiitable.getDefault())
olc.atomradiitable = CovalentRadiiTable()
olc.atomradiitable.setCustom('Na', 2)
olc.atomradiitable.foo = 123
spkl2 = pickle.dumps(olc)
olc2 = pickle.loads(spkl2)
self.assertEqual(2, olc2.atomradiitable.lookup('Na'))
self.assertEqual(1, len(olc2.atomradiitable.getAllCustom()))
self.assertEqual(123, olc2.atomradiitable.foo)
return
def setUp(self):
self.rtb = CovalentRadiiTable()
return
