def test_error_03(self):
"""Capteurs error."""
code = 'B440112201'
capteurs = (
sitehydro.Capteur(code='%s01' % code, typemesure='Q'),
sitehydro.Capteur(code='%s02' % code, typemesure='H'),
)
sitehydro.Stationhydro(**{
'code': code,
'typestation': 'DEB',
'capteurs': capteurs
})
self.assertRaises(TypeError, sitehydro.Stationhydro, **{
'code': code,
'capteurs': 'c'
})
self.assertRaises(ValueError, sitehydro.Stationhydro, **{
'code': code,
'capteurs': capteurs
})
self.assertRaises(
ValueError, sitehydro.Stationhydro, **{
'code': code,
'typestation': 'LIMNI',
'capteurs': capteurs
})
self.assertRaises(
ValueError, sitehydro.Stationhydro, **{
'code': code,
'typestation': 'HC',
'capteurs': capteurs
})
def _serie_from_element(element):
"""Return a obshydro.Serie from a <Serie> element."""
if element is not None:
# entite can be a Sitehydro, a Stationhydro or a Capteur
entite = None
if element.find('CdSiteHydro') is not None:
entite = _sitehydro.Sitehydro(
code=_value(element, 'CdSiteHydro')
)
elif element.find('CdStationHydro') is not None:
entite = _sitehydro.Stationhydro(
code=_value(element, 'CdStationHydro')
)
elif element.find('CdCapteur') is not None:
entite = _sitehydro.Capteur(
code=_value(element, 'CdCapteur')
)
# make the Serie
return _obshydro.Serie(
entite=entite,
grandeur=_value(element, 'GrdSerie'),
statut=_value(element, 'StatutSerie'),
observations=_observations_from_element(element.find('ObssHydro'))
)
def test_error_02(self):
"""Code error."""
sitehydro.Capteur(**{'code': 'B44011220101'})
self.assertRaises(TypeError, sitehydro.Capteur, **{'code': None})
self.assertRaises(ValueError, sitehydro.Capteur,
**{'code': 'B440112201'})
self.assertRaises(ValueError, sitehydro.Capteur,
**{'code': 'B4401122010133'})
def test_error_01(self):
"""Typemesure error."""
c = sitehydro.Capteur(**{'code': 'A14410010201', 'typemesure': 'H'})
self.assertRaises(TypeError, c.__setattr__, *('typemesure', None))
self.assertRaises(ValueError, sitehydro.Capteur, **{
'code': 'A14410010201',
'typemesure': 'RR'
})
def test_base_02(self):
"""Base case test."""
typemesure = 'Q'
code = 'A03346500101'
libelle = 'Capteur de secours'
c = sitehydro.Capteur(code=code,
typemesure=typemesure,
libelle=libelle)
self.assertEqual((c.code, c.typemesure, c.libelle),
(code, typemesure, libelle))
def _capteur_from_element(element):
"""Return a sitehydro.Capteur from a <Capteur> element."""
if element is not None:
# prepare args
args = {}
args['code'] = _value(element, 'CdCapteur')
args['libelle'] = _value(element, 'LbCapteur')
typemesure = _value(element, 'TypMesureCapteur')
if typemesure is not None:
args['typemesure'] = typemesure
# build Capteur
return _sitehydro.Capteur(**args)
def test_base_02(self):
"""Base case test."""
code = 'A033465001'
typestation = 'LIMNI'
libelle = 'La Seine a Paris - rive droite'
capteurs = [sitehydro.Capteur(code='V83310100101')]
s = sitehydro.Stationhydro(code=code,
typestation=typestation,
libelle=libelle,
capteurs=capteurs)
self.assertEqual((s.code, s.typestation, s.libelle, s.capteurs),
(code, typestation, libelle, capteurs))
def test_base_03(self):
"""Update capteurs attribute."""
code = 'A033465001'
typestation = 'LIMNI'
libelle = 'La Seine a Paris - rive droite'
capteurs = [sitehydro.Capteur(code='V83310100101')]
s = sitehydro.Stationhydro(code=code,
typestation=typestation,
libelle=libelle,
capteurs=capteurs)
self.assertEqual((s.code, s.typestation, s.libelle),
(code, typestation, libelle))
s.capteurs = None
self.assertEqual(s.capteurs, [])
s.capteurs = capteurs[0]
self.assertEqual(s.capteurs, capteurs)
s.capteurs = capteurs
self.assertEqual(s.capteurs, capteurs)
def test_fuzzy_mode_01(self):
"""Fuzzy mode test."""
typemesure = 'RR'
code = 'C1'
c = sitehydro.Capteur(code=code, typemesure=typemesure, strict=False)
self.assertEqual((c.code, c.typemesure), (code, typemesure))
def test_str_01(self):
"""Test __str__ method with None values."""
c = sitehydro.Capteur(code=0, strict=False)
self.assertTrue(c.__str__().rfind('Capteur') > -1)
def test_base_01(self):
"""Base case with empty capteur."""
code = 'V83310100101'
c = sitehydro.Capteur(code=code)
self.assertEqual((c.code, c.typemesure, c.libelle), (code, 'H', None))
