refactor(signal_streams): refactor signal_streams.py by applying black

Met4FoF · Jun 29, 2021 · e604418 · e604418
1 parent 11368ee
commit e604418
Showing 1 changed file with 28 additions and 6 deletions.
diff --git a/agentMET4FOF/streams/signal_streams.py b/agentMET4FOF/streams/signal_streams.py
@@ -19,10 +19,20 @@ class SineGenerator(DataStreamMET4FOF):
     sine_freq : float
         frequency of wave function
     """
+
     def __init__(self, sfreq=500, sine_freq=50):
         super().__init__()
-        self.set_metadata("SineGenerator","time","s",("Voltage"),("V"),"Simple sine wave generator")
-        self.set_generator_function(generator_function=self.sine_wave_function, sfreq=sfreq, sine_freq=sine_freq)
+        self.set_metadata(
+            "SineGenerator",
+            "time",
+            "s",
+            ("Voltage"),
+            ("V"),
+            "Simple sine wave generator",
+        )
+        self.set_generator_function(
+            generator_function=self.sine_wave_function, sfreq=sfreq, sine_freq=sine_freq
+        )
 
     def sine_wave_function(self, time, sine_freq):
         """A simple sine wave generator"""
@@ -47,12 +57,24 @@ class CosineGenerator(DataStreamMET4FOF):
     F : int
         frequency of wave function
     """
-    def __init__(self, sfreq = 500, cosine_freq=5):
+
+    def __init__(self, sfreq=500, cosine_freq=5):
         super().__init__()
-        self.set_metadata("CosineGenerator","time","s",("Voltage"),("V"),"Simple cosine wave generator")
-        self.set_generator_function(generator_function=self.cosine_wave_function, sfreq=sfreq, cosine_freq=cosine_freq)
+        self.set_metadata(
+            "CosineGenerator",
+            "time",
+            "s",
+            ("Voltage"),
+            ("V"),
+            "Simple cosine wave generator",
+        )
+        self.set_generator_function(
+            generator_function=self.cosine_wave_function,
+            sfreq=sfreq,
+            cosine_freq=cosine_freq,
+        )
 
     def cosine_wave_function(self, time, cosine_freq=50):
         """A simple cosine wave generator"""
         value = np.cos(2 * np.pi * cosine_freq * time)
-        return value
+        return value