blacken test code

1 files changed, 138 insertions, 63 deletions

diff --git a/test/test_timingharness.py b/test/test_timingharness.py
index 9c49b46..29e21f8 100755
--- a/test/test_timingharness.py
+++ b/test/test_timingharness.py
@@ -7,89 +7,164 @@ import unittest
 
 class TestModels(unittest.TestCase):
     def test_model_singlefile_rf24(self):
-        raw_data = TimingData(['test-data/20190815_111745_nRF24_no-rx.json'])
+        raw_data = TimingData(["test-data/20190815_111745_nRF24_no-rx.json"])
         preprocessed_data = raw_data.get_preprocessed_data()
         by_name, parameters, arg_count = pta_trace_to_aggregate(preprocessed_data)
         model = AnalyticModel(by_name, parameters, arg_count)
-        self.assertEqual(model.names, 'setPALevel setRetries setup write'.split(' '))
+        self.assertEqual(model.names, "setPALevel setRetries setup write".split(" "))
         static_model = model.get_static()
-        self.assertAlmostEqual(static_model('setPALevel', 'duration'), 146, places=0)
-        self.assertAlmostEqual(static_model('setRetries', 'duration'), 73, places=0)
-        self.assertAlmostEqual(static_model('setup', 'duration'), 6533, places=0)
-        self.assertAlmostEqual(static_model('write', 'duration'), 12634, places=0)
-
-        for transition in 'setPALevel setRetries setup write'.split(' '):
-            self.assertAlmostEqual(model.stats.param_dependence_ratio(transition, 'duration', 'channel'), 0, places=2)
+        self.assertAlmostEqual(static_model("setPALevel", "duration"), 146, places=0)
+        self.assertAlmostEqual(static_model("setRetries", "duration"), 73, places=0)
+        self.assertAlmostEqual(static_model("setup", "duration"), 6533, places=0)
+        self.assertAlmostEqual(static_model("write", "duration"), 12634, places=0)
+
+        for transition in "setPALevel setRetries setup write".split(" "):
+            self.assertAlmostEqual(
+                model.stats.param_dependence_ratio(transition, "duration", "channel"),
+                0,
+                places=2,
+            )
 
         param_model, param_info = model.get_fitted()
-        self.assertEqual(param_info('setPALevel', 'duration'), None)
-        self.assertEqual(param_info('setRetries', 'duration'), None)
-        self.assertEqual(param_info('setup', 'duration'), None)
-        self.assertEqual(param_info('write', 'duration')['function']._model_str, '0 + regression_arg(0) + regression_arg(1) * parameter(max_retry_count) + regression_arg(2) * parameter(retry_delay) + regression_arg(3) * parameter(max_retry_count) * parameter(retry_delay)')
-
-        self.assertAlmostEqual(param_info('write', 'duration')['function']._regression_args[0], 1163, places=0)
-        self.assertAlmostEqual(param_info('write', 'duration')['function']._regression_args[1], 464, places=0)
-        self.assertAlmostEqual(param_info('write', 'duration')['function']._regression_args[2], 1, places=0)
-        self.assertAlmostEqual(param_info('write', 'duration')['function']._regression_args[3], 1, places=0)
+        self.assertEqual(param_info("setPALevel", "duration"), None)
+        self.assertEqual(param_info("setRetries", "duration"), None)
+        self.assertEqual(param_info("setup", "duration"), None)
+        self.assertEqual(
+            param_info("write", "duration")["function"]._model_str,
+            "0 + regression_arg(0) + regression_arg(1) * parameter(max_retry_count) + regression_arg(2) * parameter(retry_delay) + regression_arg(3) * parameter(max_retry_count) * parameter(retry_delay)",
+        )
+
+        self.assertAlmostEqual(
+            param_info("write", "duration")["function"]._regression_args[0],
+            1163,
+            places=0,
+        )
+        self.assertAlmostEqual(
+            param_info("write", "duration")["function"]._regression_args[1],
+            464,
+            places=0,
+        )
+        self.assertAlmostEqual(
+            param_info("write", "duration")["function"]._regression_args[2], 1, places=0
+        )
+        self.assertAlmostEqual(
+            param_info("write", "duration")["function"]._regression_args[3], 1, places=0
+        )
 
     def test_dependent_parameter_pruning(self):
-        raw_data = TimingData(['test-data/20190815_103347_nRF24_no-rx.json'])
+        raw_data = TimingData(["test-data/20190815_103347_nRF24_no-rx.json"])
         preprocessed_data = raw_data.get_preprocessed_data()
         by_name, parameters, arg_count = pta_trace_to_aggregate(preprocessed_data)
         prune_dependent_parameters(by_name, parameters)
         model = AnalyticModel(by_name, parameters, arg_count)
-        self.assertEqual(model.names, 'getObserveTx setPALevel setRetries setup write'.split(' '))
+        self.assertEqual(
+            model.names, "getObserveTx setPALevel setRetries setup write".split(" ")
+        )
         static_model = model.get_static()
-        self.assertAlmostEqual(static_model('getObserveTx', 'duration'), 75, places=0)
-        self.assertAlmostEqual(static_model('setPALevel', 'duration'), 146, places=0)
-        self.assertAlmostEqual(static_model('setRetries', 'duration'), 73, places=0)
-        self.assertAlmostEqual(static_model('setup', 'duration'), 6533, places=0)
-        self.assertAlmostEqual(static_model('write', 'duration'), 12634, places=0)
-
-        for transition in 'getObserveTx setPALevel setRetries setup write'.split(' '):
-            self.assertAlmostEqual(model.stats.param_dependence_ratio(transition, 'duration', 'channel'), 0, places=2)
+        self.assertAlmostEqual(static_model("getObserveTx", "duration"), 75, places=0)
+        self.assertAlmostEqual(static_model("setPALevel", "duration"), 146, places=0)
+        self.assertAlmostEqual(static_model("setRetries", "duration"), 73, places=0)
+        self.assertAlmostEqual(static_model("setup", "duration"), 6533, places=0)
+        self.assertAlmostEqual(static_model("write", "duration"), 12634, places=0)
+
+        for transition in "getObserveTx setPALevel setRetries setup write".split(" "):
+            self.assertAlmostEqual(
+                model.stats.param_dependence_ratio(transition, "duration", "channel"),
+                0,
+                places=2,
+            )
 
         param_model, param_info = model.get_fitted()
-        self.assertEqual(param_info('getObserveTx', 'duration'), None)
-        self.assertEqual(param_info('setPALevel', 'duration'), None)
-        self.assertEqual(param_info('setRetries', 'duration'), None)
-        self.assertEqual(param_info('setup', 'duration'), None)
-        self.assertEqual(param_info('write', 'duration')['function']._model_str, '0 + regression_arg(0) + regression_arg(1) * parameter(max_retry_count) + regression_arg(2) * parameter(retry_delay) + regression_arg(3) * parameter(max_retry_count) * parameter(retry_delay)')
-
-        self.assertAlmostEqual(param_info('write', 'duration')['function']._regression_args[0], 1163, places=0)
-        self.assertAlmostEqual(param_info('write', 'duration')['function']._regression_args[1], 464, places=0)
-        self.assertAlmostEqual(param_info('write', 'duration')['function']._regression_args[2], 1, places=0)
-        self.assertAlmostEqual(param_info('write', 'duration')['function']._regression_args[3], 1, places=0)
+        self.assertEqual(param_info("getObserveTx", "duration"), None)
+        self.assertEqual(param_info("setPALevel", "duration"), None)
+        self.assertEqual(param_info("setRetries", "duration"), None)
+        self.assertEqual(param_info("setup", "duration"), None)
+        self.assertEqual(
+            param_info("write", "duration")["function"]._model_str,
+            "0 + regression_arg(0) + regression_arg(1) * parameter(max_retry_count) + regression_arg(2) * parameter(retry_delay) + regression_arg(3) * parameter(max_retry_count) * parameter(retry_delay)",
+        )
+
+        self.assertAlmostEqual(
+            param_info("write", "duration")["function"]._regression_args[0],
+            1163,
+            places=0,
+        )
+        self.assertAlmostEqual(
+            param_info("write", "duration")["function"]._regression_args[1],
+            464,
+            places=0,
+        )
+        self.assertAlmostEqual(
+            param_info("write", "duration")["function"]._regression_args[2], 1, places=0
+        )
+        self.assertAlmostEqual(
+            param_info("write", "duration")["function"]._regression_args[3], 1, places=0
+        )
 
     def test_function_override(self):
-        raw_data = TimingData(['test-data/20190815_122531_nRF24_no-rx.json'])
+        raw_data = TimingData(["test-data/20190815_122531_nRF24_no-rx.json"])
         preprocessed_data = raw_data.get_preprocessed_data()
         by_name, parameters, arg_count = pta_trace_to_aggregate(preprocessed_data)
-        model = AnalyticModel(by_name, parameters, arg_count, function_override={('write', 'duration'): '(parameter(auto_ack!) * (regression_arg(0) + regression_arg(1) * parameter(max_retry_count) + regression_arg(2) * parameter(retry_delay) + regression_arg(3) * parameter(max_retry_count) * parameter(retry_delay))) + ((1 - parameter(auto_ack!)) * regression_arg(4))'})
-        self.assertEqual(model.names, 'setAutoAck setPALevel setRetries setup write'.split(' '))
+        model = AnalyticModel(
+            by_name,
+            parameters,
+            arg_count,
+            function_override={
+                (
+                    "write",
+                    "duration",
+                ): "(parameter(auto_ack!) * (regression_arg(0) + regression_arg(1) * parameter(max_retry_count) + regression_arg(2) * parameter(retry_delay) + regression_arg(3) * parameter(max_retry_count) * parameter(retry_delay))) + ((1 - parameter(auto_ack!)) * regression_arg(4))"
+            },
+        )
+        self.assertEqual(
+            model.names, "setAutoAck setPALevel setRetries setup write".split(" ")
+        )
         static_model = model.get_static()
-        self.assertAlmostEqual(static_model('setAutoAck', 'duration'), 72, places=0)
-        self.assertAlmostEqual(static_model('setPALevel', 'duration'), 146, places=0)
-        self.assertAlmostEqual(static_model('setRetries', 'duration'), 73, places=0)
-        self.assertAlmostEqual(static_model('setup', 'duration'), 6533, places=0)
-        self.assertAlmostEqual(static_model('write', 'duration'), 1181, places=0)
-
-        for transition in 'setAutoAck setPALevel setRetries setup write'.split(' '):
-            self.assertAlmostEqual(model.stats.param_dependence_ratio(transition, 'duration', 'channel'), 0, places=2)
+        self.assertAlmostEqual(static_model("setAutoAck", "duration"), 72, places=0)
+        self.assertAlmostEqual(static_model("setPALevel", "duration"), 146, places=0)
+        self.assertAlmostEqual(static_model("setRetries", "duration"), 73, places=0)
+        self.assertAlmostEqual(static_model("setup", "duration"), 6533, places=0)
+        self.assertAlmostEqual(static_model("write", "duration"), 1181, places=0)
+
+        for transition in "setAutoAck setPALevel setRetries setup write".split(" "):
+            self.assertAlmostEqual(
+                model.stats.param_dependence_ratio(transition, "duration", "channel"),
+                0,
+                places=2,
+            )
 
         param_model, param_info = model.get_fitted()
-        self.assertEqual(param_info('setAutoAck', 'duration'), None)
-        self.assertEqual(param_info('setPALevel', 'duration'), None)
-        self.assertEqual(param_info('setRetries', 'duration'), None)
-        self.assertEqual(param_info('setup', 'duration'), None)
-        self.assertEqual(param_info('write', 'duration')['function']._model_str, '(parameter(auto_ack!) * (regression_arg(0) + regression_arg(1) * parameter(max_retry_count) + regression_arg(2) * parameter(retry_delay) + regression_arg(3) * parameter(max_retry_count) * parameter(retry_delay))) + ((1 - parameter(auto_ack!)) * regression_arg(4))')
-
-        self.assertAlmostEqual(param_info('write', 'duration')['function']._regression_args[0], 1162, places=0)
-        self.assertAlmostEqual(param_info('write', 'duration')['function']._regression_args[1], 464, places=0)
-        self.assertAlmostEqual(param_info('write', 'duration')['function']._regression_args[2], 1, places=0)
-        self.assertAlmostEqual(param_info('write', 'duration')['function']._regression_args[3], 1, places=0)
-        self.assertAlmostEqual(param_info('write', 'duration')['function']._regression_args[4], 1086, places=0)
-
-
-if __name__ == '__main__':
+        self.assertEqual(param_info("setAutoAck", "duration"), None)
+        self.assertEqual(param_info("setPALevel", "duration"), None)
+        self.assertEqual(param_info("setRetries", "duration"), None)
+        self.assertEqual(param_info("setup", "duration"), None)
+        self.assertEqual(
+            param_info("write", "duration")["function"]._model_str,
+            "(parameter(auto_ack!) * (regression_arg(0) + regression_arg(1) * parameter(max_retry_count) + regression_arg(2) * parameter(retry_delay) + regression_arg(3) * parameter(max_retry_count) * parameter(retry_delay))) + ((1 - parameter(auto_ack!)) * regression_arg(4))",
+        )
+
+        self.assertAlmostEqual(
+            param_info("write", "duration")["function"]._regression_args[0],
+            1162,
+            places=0,
+        )
+        self.assertAlmostEqual(
+            param_info("write", "duration")["function"]._regression_args[1],
+            464,
+            places=0,
+        )
+        self.assertAlmostEqual(
+            param_info("write", "duration")["function"]._regression_args[2], 1, places=0
+        )
+        self.assertAlmostEqual(
+            param_info("write", "duration")["function"]._regression_args[3], 1, places=0
+        )
+        self.assertAlmostEqual(
+            param_info("write", "duration")["function"]._regression_args[4],
+            1086,
+            places=0,
+        )
+
+
+if __name__ == "__main__":
     unittest.main()