Skip to content

Commit 0832b07

Browse files
committed
TST: exceptions for invalid setPairMask arguments
And verify it can accept numpy.int.
1 parent f05127a commit 0832b07

1 file changed

Lines changed: 17 additions & 0 deletions

File tree

src/diffpy/srreal/tests/testpairquantity.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55

66
import unittest
77
import pickle
8+
import numpy
89

910
from diffpy.srreal.pairquantity import PairQuantity
1011
from diffpy.srreal.srreal_ext import BasePairQuantity
@@ -76,6 +77,22 @@ def test_setStructure(self):
7677
return
7778

7879

80+
def test_setPairMask_args(self):
81+
"""check argument type handling in setPairMask
82+
"""
83+
spm = self.pq.setPairMask
84+
gpm = self.pq.getPairMask
85+
self.assertRaises(TypeError, spm, 0.0, 0, False)
86+
self.assertRaises(TypeError, spm, numpy.float32(0.5), 0, False)
87+
self.assertTrue(gpm(0, 0))
88+
spm(numpy.int32(1), 0, True, others=False)
89+
self.assertTrue(gpm(0, 1))
90+
self.assertTrue(gpm(1, 0))
91+
self.assertFalse(gpm(0, 0))
92+
self.assertFalse(gpm(2, 7))
93+
return
94+
95+
7996
def test_getStructure(self):
8097
"""check PairQuantity.getStructure()
8198
"""

0 commit comments

Comments
 (0)