[tor-commits] [sbws/master] generate: use round_digs variable name in methods

juga at torproject.org juga at torproject.org
Fri Dec 21 08:54:09 UTC 2018


commit c57ba63ed0107e2ae07271302cfa7c9530774043
Author: juga0 <juga at riseup.net>
Date:   Fri Nov 23 17:40:23 2018 +0000

    generate: use round_digs variable name in methods
    
    The torflow-round-digs cli argument was kept, but it was added
    other argument, round-digs, for the same option.
    argparse takes the first argument as the variable name, but the
    methods where still using torflow_round_digs.
    
    Closes bug #28602. Bugfix 1.0.3-dev0.
---
 sbws/core/generate.py           |  2 +-
 sbws/lib/v3bwfile.py            |  4 ++--
 tests/unit/lib/test_v3bwfile.py | 21 ++++++++++++++-------
 3 files changed, 17 insertions(+), 10 deletions(-)

diff --git a/sbws/core/generate.py b/sbws/core/generate.py
index f9b46cb..971ed9e 100644
--- a/sbws/core/generate.py
+++ b/sbws/core/generate.py
@@ -106,7 +106,7 @@ def main(args, conf):
     bw_file = V3BWFile.from_results(results, state_fpath, args.scale_constant,
                                     scaling_method,
                                     torflow_cap=args.torflow_bw_margin,
-                                    torflow_round_digs=args.torflow_round_digs,
+                                    round_digs=args.round_digs,
                                     secs_recent=args.secs_recent,
                                     secs_away=args.secs_away,
                                     min_num=args.min_num,
diff --git a/sbws/lib/v3bwfile.py b/sbws/lib/v3bwfile.py
index a050744..d0e53f9 100644
--- a/sbws/lib/v3bwfile.py
+++ b/sbws/lib/v3bwfile.py
@@ -488,7 +488,7 @@ class V3BWFile(object):
                      scaling_method=TORFLOW_SCALING,
                      torflow_obs=TORFLOW_OBS_LAST,
                      torflow_cap=TORFLOW_BW_MARGIN,
-                     torflow_round_digs=PROP276_ROUND_DIG,
+                     round_digs=PROP276_ROUND_DIG,
                      secs_recent=None, secs_away=None, min_num=0,
                      consensus_path=None, max_bw_diff_perc=MAX_BW_DIFF_PERC,
                      reverse=False):
@@ -533,7 +533,7 @@ class V3BWFile(object):
             # log.debug(bw_lines[-1])
         elif scaling_method == TORFLOW_SCALING:
             bw_lines = cls.bw_torflow_scale(bw_lines_raw, torflow_obs,
-                                            torflow_cap, torflow_round_digs)
+                                            torflow_cap, round_digs)
             # log.debug(bw_lines[-1])
             cls.update_progress(
                 cls, bw_lines, header, number_consensus_relays, state)
diff --git a/tests/unit/lib/test_v3bwfile.py b/tests/unit/lib/test_v3bwfile.py
index f9c50d5..86b979b 100644
--- a/tests/unit/lib/test_v3bwfile.py
+++ b/tests/unit/lib/test_v3bwfile.py
@@ -6,7 +6,7 @@ import os.path
 
 from sbws import __version__ as version
 from sbws.globals import (SPEC_VERSION, SBWS_SCALING, TORFLOW_SCALING,
-                          MIN_REPORT)
+                          MIN_REPORT, TORFLOW_ROUND_DIG, PROP276_ROUND_DIG)
 from sbws.lib.resultdump import Result, load_result_file, ResultSuccess
 from sbws.lib.v3bwfile import (V3BWHeader, V3BWLine, TERMINATOR, LINE_SEP,
                                KEYVALUE_SEP_V1, num_results_of_type,
@@ -266,14 +266,21 @@ def test_sbws_scale(datadir):
 
 def test_torflow_scale(datadir):
     results = load_result_file(str(datadir.join("results.txt")))
-    v3bwfile = V3BWFile.from_results(results, scaling_method=TORFLOW_SCALING)
-    assert v3bwfile.bw_lines[0].bw == 520
     v3bwfile = V3BWFile.from_results(results, scaling_method=TORFLOW_SCALING,
-                                     torflow_cap=0.0001)
-    assert v3bwfile.bw_lines[0].bw == 520
+                                     round_digs=TORFLOW_ROUND_DIG)
+    assert v3bwfile.bw_lines[0].bw == 524
+    v3bwfile = V3BWFile.from_results(results, scaling_method=TORFLOW_SCALING,
+                                     torflow_cap=0.0001,
+                                     round_digs=TORFLOW_ROUND_DIG)
+    assert v3bwfile.bw_lines[0].bw == 524
     v3bwfile = V3BWFile.from_results(results, scaling_method=TORFLOW_SCALING,
-                                     torflow_cap=1, torflow_round_digs=1)
-    assert v3bwfile.bw_lines[0].bw == 500
+                                     torflow_cap=1,
+                                     round_digs=TORFLOW_ROUND_DIG)
+    assert v3bwfile.bw_lines[0].bw == 524
+    v3bwfile = V3BWFile.from_results(results, scaling_method=TORFLOW_SCALING,
+                                     torflow_cap=1,
+                                     round_digs=PROP276_ROUND_DIG)
+    assert v3bwfile.bw_lines[0].bw == 520
 
 
 def test_results_away_each_other(datadir):





More information about the tor-commits mailing list