def test_DiskElectrode():
with pytest.raises(TypeError):
DiskElectrode(0, 0, 0, [1, 2])
with pytest.raises(TypeError):
DiskElectrode(0, np.array([0, 1]), 0, 1)
# Invalid radius:
with pytest.raises(ValueError):
DiskElectrode(0, 0, 0, -5)
# Check params:
electrode = DiskElectrode(0, 1, 2, 100)
npt.assert_almost_equal(electrode.x, 0)
npt.assert_almost_equal(electrode.y, 1)
npt.assert_almost_equal(electrode.z, 2)
# On the electrode surface (z=2, x^2+y^2<=100^2)
npt.assert_almost_equal(electrode.electric_potential(0, 1, 2, 1), 1)
npt.assert_almost_equal(electrode.electric_potential(30, -30, 2, 1), 1)
npt.assert_almost_equal(electrode.electric_potential(0, 101, 2, 1), 1)
npt.assert_almost_equal(electrode.electric_potential(0, -99, 2, 1), 1)
npt.assert_almost_equal(electrode.electric_potential(100, 1, 2, 1), 1)
npt.assert_almost_equal(electrode.electric_potential(-100, 1, 2, 1), 1)
# Right off the surface (z=2, x^2+y^2>100^2)
npt.assert_almost_equal(electrode.electric_potential(0, 102, 2, 1),
0.910,
decimal=3)
npt.assert_almost_equal(electrode.electric_potential(0, -100, 2, 1),
0.910,
decimal=3)
# Some distance away from the electrode (z>2):
npt.assert_almost_equal(electrode.electric_potential(0, 1, 38, 1),
0.780,
decimal=3)
def test_DiskElectrode():
with pytest.raises(TypeError):
DiskElectrode(0, 0, 0, [1, 2])
with pytest.raises(TypeError):
DiskElectrode(0, np.array([0, 1]), 0, 1)
# Invalid radius:
with pytest.raises(ValueError):
DiskElectrode(0, 0, 0, -5)
# Check params:
electrode = DiskElectrode(0, 1, 2, 100, name='A001')
npt.assert_almost_equal(electrode.x, 0)
npt.assert_almost_equal(electrode.y, 1)
npt.assert_almost_equal(electrode.z, 2)
npt.assert_equal(electrode.name, 'A001')
# On the electrode surface (z=2, x^2+y^2<=100^2)
npt.assert_almost_equal(electrode.electric_potential(0, 1, 2, 1), 1)
npt.assert_almost_equal(electrode.electric_potential(30, -30, 2, 1), 1)
npt.assert_almost_equal(electrode.electric_potential(0, 101, 2, 1), 1)
npt.assert_almost_equal(electrode.electric_potential(0, -99, 2, 1), 1)
npt.assert_almost_equal(electrode.electric_potential(100, 1, 2, 1), 1)
npt.assert_almost_equal(electrode.electric_potential(-100, 1, 2, 1), 1)
# Right off the surface (z=2, x^2+y^2>100^2)
npt.assert_almost_equal(electrode.electric_potential(0, 102, 2, 1),
0.910,
decimal=3)
npt.assert_almost_equal(electrode.electric_potential(0, -100, 2, 1),
0.910,
decimal=3)
# Some distance away from the electrode (z>2):
npt.assert_almost_equal(electrode.electric_potential(0, 1, 38, 1),
0.780,
decimal=3)
# Slots:
npt.assert_equal(hasattr(electrode, '__slots__'), True)
npt.assert_equal(hasattr(electrode, '__dict__'), False)
# Plots:
ax = electrode.plot()
npt.assert_equal(len(ax.texts), 0)
npt.assert_equal(len(ax.patches), 1)
npt.assert_equal(isinstance(ax.patches[0], Circle), True)