def test_similarity_dist(self): subject = from_file('floodestimation/tests/data/37017.CD3') donor = from_file('floodestimation/tests/data/37020.CD3') analysis = GrowthCurveAnalysis(subject) result = analysis._similarity_distance(subject, donor) expected = 0.1159 # Science Report SC050050, table 6.6, row 2 self.assertAlmostEqual(result, expected, places=4)
def test_37017(self): subject = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(subject) self.assertEqual(len(subject.amax_records), 34) var, skew = analysis._var_and_skew(subject) self.assertAlmostEqual(var, 0.2232, places=4) self.assertAlmostEqual(skew, -0.0908, places=4)
def test_similarity_dist(self): subject = from_file('floodestimation/tests/data/37017.CD3') donor = from_file('floodestimation/tests/data/37020.CD3') analysis = GrowthCurveAnalysis(subject) result = analysis._similarity_distance(subject, donor) expected = 0.1159 # Science Report SC050050, table 6.6, row 2 self.assertAlmostEqual(result, expected, places=4)
def test_37017(self): subject = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(subject) self.assertEqual(len(subject.amax_records), 34) var, skew = analysis._var_and_skew(subject) self.assertAlmostEqual(var, 0.2232, places=4) self.assertAlmostEqual(skew, -0.0908, places=4)
def test_l_skew_weight(self): subject = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(subject) donor = copy(subject) donor.similarity_dist = 0.2010 result = analysis._l_skew_weight(donor) expected = 47.34 # Science Report SC050050, table 6.6, row 4 (note that donor has same record length as subject) self.assertAlmostEqual(result, expected, places=1)
def test_dimensionless_flows(self): analysis = GrowthCurveAnalysis(self.catchment) self.catchment.amax_records = [AmaxRecord(date(1999, 12, 31), 3.0, 0.5), AmaxRecord(date(2000, 12, 31), 2.0, 0.5), AmaxRecord(date(2001, 12, 31), 1.0, 0.5)] result = analysis._dimensionless_flows(self.catchment) expected = np.array([1.5, 1, 0.5]) assert_almost_equal(result, expected)
def test_l_skew_weight(self): subject = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(subject) donor = copy(subject) donor.similarity_dist = 0.2010 result = analysis._l_skew_weight(donor) expected = 47.34 # Science Report SC050050, table 6.6, row 4 (note that donor has same record length as subject) self.assertAlmostEqual(result, expected, places=1)
def test_l_cv_and_skew_one_donor_urban(self): catchment = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(catchment, year=2000) analysis.donor_catchments = [catchment] var, skew = analysis._var_and_skew(analysis.donor_catchments) self.assertAlmostEqual(var, 0.21987, places=4) self.assertAlmostEqual(skew, -0.08746, places=4)
def test_l_cv_and_skew_one_donor_urban(self): catchment = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(catchment, year=2000) analysis.donor_catchments = [catchment] var, skew = analysis._var_and_skew(analysis.donor_catchments) self.assertAlmostEqual(var, 0.21987, places=4) self.assertAlmostEqual(skew, -0.08746, places=4)
def test_similarity_distance_incomplete_descriptors(self): other_catchment = Catchment(location="Burn A", watercourse="Village B") other_catchment.id = 999 other_catchment.is_suitable_for_pooling = True self.db_session.add(other_catchment) gauged_catchments = CatchmentCollections(self.db_session) analysis = GrowthCurveAnalysis(self.catchment, gauged_catchments) self.assertEqual(float('inf'), analysis._similarity_distance(self.catchment, other_catchment))
def test_l_cv_and_skew(self): gauged_catchments = CatchmentCollections(self.db_session) catchment = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(catchment, gauged_catchments) var, skew = analysis._var_and_skew(catchment) self.assertAlmostEqual(var, 0.2232, places=4) self.assertAlmostEqual(skew, -0.0908, places=4)
def test_l_cv_and_skew(self): gauged_catchments = CatchmentCollections(self.db_session) catchment = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(catchment, gauged_catchments) var, skew = analysis._var_and_skew(catchment) self.assertAlmostEqual(var, 0.2232, places=4) self.assertAlmostEqual(skew, -0.0908, places=4)
def test_dimensionless_flows(self): analysis = GrowthCurveAnalysis(self.catchment) self.catchment.amax_records = [ AmaxRecord(date(1999, 12, 31), 3.0, 0.5), AmaxRecord(date(2000, 12, 31), 2.0, 0.5), AmaxRecord(date(2001, 12, 31), 1.0, 0.5) ] result = analysis._dimensionless_flows(self.catchment) expected = np.array([1.5, 1, 0.5]) assert_almost_equal(result, expected)
def test_similarity_distance_incomplete_descriptors(self): other_catchment = Catchment(location="Burn A", watercourse="Village B") other_catchment.id = 999 other_catchment.is_suitable_for_pooling = True self.db_session.add(other_catchment) gauged_catchments = CatchmentCollections(self.db_session) analysis = GrowthCurveAnalysis(self.catchment, gauged_catchments) self.assertEqual( float('inf'), analysis._similarity_distance(self.catchment, other_catchment))
def test_find_donors_exclude_urban(self): other_catchment = Catchment(location="Burn A", watercourse="Village B") other_catchment.id = 999 other_catchment.is_suitable_for_pooling = True other_catchment.descriptors = Descriptors(urbext2000=0.031) self.db_session.add(other_catchment) gauged_catchments = CatchmentCollections(self.db_session) analysis = GrowthCurveAnalysis(self.catchment, gauged_catchments) analysis.find_donor_catchments() donor_ids = [d.id for d in analysis.donor_catchments] self.assertEqual([10002, 10001], donor_ids)
def test_find_donors_exclude_urban(self): other_catchment = Catchment(location="Burn A", watercourse="Village B") other_catchment.id = 999 other_catchment.is_suitable_for_pooling = True other_catchment.descriptors = Descriptors(urbext2000=0.031) self.db_session.add(other_catchment) gauged_catchments = CatchmentCollections(self.db_session) analysis = GrowthCurveAnalysis(self.catchment, gauged_catchments) analysis.find_donor_catchments() donor_ids = [d.id for d in analysis.donor_catchments] self.assertEqual([10002, 10001], donor_ids)
def test_dist_params(self): catchment = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(catchment) growth_curve = analysis.growth_curve(method='single_site') params = growth_curve.params self.assertAlmostEqual(params['loc'], 1) self.assertAlmostEqual(params['scale'], 0.2202, places=4) self.assertAlmostEqual(params['k'], 0.0908, places=4) # Kurtosis calculated from L-skew using eqns on # http://researcher.watson.ibm.com/researcher/view_group.php?id=2639 self.assertAlmostEqual(growth_curve.distr_kurtosis, 0.1735, places=4)
def test_dist_params(self): catchment = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(catchment) growth_curve = analysis.growth_curve(method='single_site') params = growth_curve.params self.assertAlmostEqual(params['loc'], 1) self.assertAlmostEqual(params['scale'], 0.2202, places=4) self.assertAlmostEqual(params['k'], 0.0908, places=4) # Kurtosis calculated from L-skew using eqns on # http://researcher.watson.ibm.com/researcher/view_group.php?id=2639 self.assertAlmostEqual(growth_curve.distr_kurtosis, 0.1735, places=4)
def run_growthcurve(self): results = {} analysis = GrowthCurveAnalysis(self.catchment, self.gauged_catchments, results_log=results) gc = analysis.growth_curve() aeps = [0.5, 0.2, 0.1, 0.05, 0.03333, 0.02, 0.01333, 0.01, 0.005, 0.002, 0.001] growth_factors = gc(aeps) flows = growth_factors * self.qmed results['aeps'] = aeps results['growth_factors'] = growth_factors results['flows'] = flows self.results['gc'] = results
def _run_growthcurve(self): results = {} analysis = GrowthCurveAnalysis(self.catchment, self.gauged_catchments, results_log=results) gc = analysis.growth_curve() aeps = [ 0.5, 0.2, 0.1, 0.05, 0.03333, 0.02, 0.01333, 0.01, 0.005, 0.002, 0.001 ] growth_factors = gc(aeps) flows = growth_factors * self.qmed results['aeps'] = aeps results['growth_factors'] = growth_factors results['flows'] = flows self.results['gc'] = results
def test_find_donors(self): gauged_catchments = CatchmentCollections(self.db_session) analysis = GrowthCurveAnalysis(self.catchment, gauged_catchments) analysis.find_donor_catchments() donor_ids = [d.id for d in analysis.donor_catchments] self.assertEqual([10002, 10001], donor_ids)
def test_single_site_gev(self): gauged_catchments = CatchmentCollections(self.db_session) catchment = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(catchment, gauged_catchments) dist_func = analysis.growth_curve(method='single_site', distr='gev') self.assertAlmostEqual(dist_func(0.5), 1)
def test_find_donors(self): gauged_catchments = CatchmentCollections(self.db_session) analysis = GrowthCurveAnalysis(self.catchment, gauged_catchments) analysis.find_donor_catchments() donor_ids = [d.id for d in analysis.donor_catchments] self.assertEqual([10002, 10001], donor_ids)
def test_single_site_gev(self): gauged_catchments = CatchmentCollections(self.db_session) catchment = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(catchment, gauged_catchments) dist_func = analysis.growth_curve(method='single_site', distr='gev') self.assertAlmostEqual(dist_func(0.5), 1)
def test_find_donors_without_collection(self): analysis = GrowthCurveAnalysis(self.catchment) self.assertFalse(analysis.find_donor_catchments())
def test_dist_param_location(self): catchment = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(catchment) growth_curve = analysis.growth_curve(method='single_site') self.assertAlmostEqual(growth_curve(0.5), 1)
def test_l_skew_weight_same_catchment(self): subject = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(subject) result = analysis._l_skew_weight(subject) expected = 116.66 # Science Report SC050050, table 6.6, row 1 self.assertAlmostEqual(result, expected, places=1)
def test_dist_param_location(self): catchment = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(catchment) growth_curve = analysis.growth_curve(method='single_site') self.assertAlmostEqual(growth_curve(0.5), 1)
def test_l_skew_weight_same_catchment(self): subject = from_file('floodestimation/tests/data/37017.CD3') analysis = GrowthCurveAnalysis(subject) result = analysis._l_skew_weight(subject) expected = 116.66 # Science Report SC050050, table 6.6, row 1 self.assertAlmostEqual(result, expected, places=1)
def test_find_donors_without_collection(self): analysis = GrowthCurveAnalysis(self.catchment) self.assertFalse(analysis.find_donor_catchments())