Skip to content

Commit c5f7611

Browse files
Update get_ionic_radii test
1 parent a7be1ff commit c5f7611

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

pyrolite/geochem/ind.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -429,7 +429,7 @@ def get_ionic_radii(
429429
430430
Returns
431431
--------
432-
:class:`pandas.Series` | :class:`float`
432+
:class:`pandas.Series` | :class:`numpy.ndarray` | :class:`float`
433433
Series with viable ion charge and coordination, with associated radii in
434434
angstroms. If the ion charge and coordiation are completely specified and
435435
found in the table, a single value will be returned instead.

test/geochem/geochem_ind.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import unittest
22

3+
import numpy as np
34
import periodictable as pt
45

56
from pyrolite.geochem.ind import (
@@ -258,11 +259,11 @@ def test_ree_radii(self):
258259

259260
def test_ree_radii_list(self):
260261
radii = get_ionic_radii(self.ree, charge=3, coordination=8)
261-
self.assertTrue(isinstance(radii, list))
262+
self.assertTrue(isinstance(radii, np.ndarray))
262263

263264
def test_ree_radii_list_whittaker_muntus(self):
264265
radii = get_ionic_radii(self.ree, charge=3, coordination=8, source="whittaker")
265-
self.assertTrue(isinstance(radii, list))
266+
self.assertTrue(isinstance(radii, np.ndarray))
266267

267268

268269
class TestByIncompatibility(unittest.TestCase):

0 commit comments

Comments
 (0)