Skip to content
Snippets Groups Projects
Commit 1baba58a authored by Oscar Gustafsson's avatar Oscar Gustafsson :bicyclist:
Browse files

Improve typing and fix typos

parent f929ba05
No related branches found
No related tags found
1 merge request!213Mixfixes
""" """
B-ASIC signal generators B-ASIC Signal Generator Module.
These can be used as input to Simulation to algorithmically provide signal values. These can be used as input to Simulation to algorithmically provide signal values.
Especially, all classes defined here will act as a callable which accepts an integer Especially, all classes defined here will act as a callable which accepts an integer
...@@ -22,7 +22,7 @@ class SignalGenerator: ...@@ -22,7 +22,7 @@ class SignalGenerator:
""" """
Base class for signal generators. Base class for signal generators.
Handles operator overloading and defined the ``__call__`` method that should Handles operator overloading and defines the ``__call__`` method that should
be overridden. be overridden.
""" """
...@@ -94,7 +94,7 @@ class Impulse(SignalGenerator): ...@@ -94,7 +94,7 @@ class Impulse(SignalGenerator):
def __call__(self, time: int) -> complex: def __call__(self, time: int) -> complex:
return 1 if time == self._delay else 0 return 1 if time == self._delay else 0
def __repr__(self): def __repr__(self) -> str:
return f"Impulse({self._delay})" if self._delay else "Impulse()" return f"Impulse({self._delay})" if self._delay else "Impulse()"
...@@ -114,7 +114,7 @@ class Step(SignalGenerator): ...@@ -114,7 +114,7 @@ class Step(SignalGenerator):
def __call__(self, time: int) -> complex: def __call__(self, time: int) -> complex:
return 1 if time >= self._delay else 0 return 1 if time >= self._delay else 0
def __repr__(self): def __repr__(self) -> str:
return f"Step({self._delay})" if self._delay else "Step()" return f"Step({self._delay})" if self._delay else "Step()"
...@@ -134,7 +134,7 @@ class Constant(SignalGenerator): ...@@ -134,7 +134,7 @@ class Constant(SignalGenerator):
def __call__(self, time: int) -> complex: def __call__(self, time: int) -> complex:
return self._constant return self._constant
def __str__(self): def __str__(self) -> str:
return f"{self._constant}" return f"{self._constant}"
...@@ -157,7 +157,7 @@ class ZeroPad(SignalGenerator): ...@@ -157,7 +157,7 @@ class ZeroPad(SignalGenerator):
return self._data[time] return self._data[time]
return 0.0 return 0.0
def __repr__(self): def __repr__(self) -> str:
return f"ZeroPad({self._data})" return f"ZeroPad({self._data})"
...@@ -181,7 +181,7 @@ class Sinusoid(SignalGenerator): ...@@ -181,7 +181,7 @@ class Sinusoid(SignalGenerator):
def __call__(self, time: int) -> complex: def __call__(self, time: int) -> complex:
return sin(pi * (self._frequency * time + self._phase)) return sin(pi * (self._frequency * time + self._phase))
def __repr__(self): def __repr__(self) -> str:
return ( return (
f"Sinusoid({self._frequency}, {self._phase})" f"Sinusoid({self._frequency}, {self._phase})"
if self._phase if self._phase
...@@ -216,7 +216,7 @@ class Gaussian(SignalGenerator): ...@@ -216,7 +216,7 @@ class Gaussian(SignalGenerator):
def __call__(self, time: int) -> complex: def __call__(self, time: int) -> complex:
return self._rng.normal(self._loc, self._scale) return self._rng.normal(self._loc, self._scale)
def __repr__(self): def __repr__(self) -> str:
ret_list = [] ret_list = []
if self._seed is not None: if self._seed is not None:
ret_list.append(f"seed={self._seed}") ret_list.append(f"seed={self._seed}")
...@@ -256,7 +256,7 @@ class Uniform(SignalGenerator): ...@@ -256,7 +256,7 @@ class Uniform(SignalGenerator):
def __call__(self, time: int) -> complex: def __call__(self, time: int) -> complex:
return self._rng.uniform(self._low, self._high) return self._rng.uniform(self._low, self._high)
def __repr__(self): def __repr__(self) -> str:
ret_list = [] ret_list = []
if self._seed is not None: if self._seed is not None:
ret_list.append(f"seed={self._seed}") ret_list.append(f"seed={self._seed}")
...@@ -280,7 +280,7 @@ class _AddGenerator(SignalGenerator): ...@@ -280,7 +280,7 @@ class _AddGenerator(SignalGenerator):
def __call__(self, time: int) -> complex: def __call__(self, time: int) -> complex:
return self._a(time) + self._b(time) return self._a(time) + self._b(time)
def __repr__(self): def __repr__(self) -> str:
return f"{self._a} + {self._b}" return f"{self._a} + {self._b}"
...@@ -296,7 +296,7 @@ class _SubGenerator(SignalGenerator): ...@@ -296,7 +296,7 @@ class _SubGenerator(SignalGenerator):
def __call__(self, time: int) -> complex: def __call__(self, time: int) -> complex:
return self._a(time) - self._b(time) return self._a(time) - self._b(time)
def __repr__(self): def __repr__(self) -> str:
return f"{self._a} - {self._b}" return f"{self._a} - {self._b}"
...@@ -312,7 +312,7 @@ class _MulGenerator(SignalGenerator): ...@@ -312,7 +312,7 @@ class _MulGenerator(SignalGenerator):
def __call__(self, time: int) -> complex: def __call__(self, time: int) -> complex:
return self._a(time) * self._b(time) return self._a(time) * self._b(time)
def __repr__(self): def __repr__(self) -> str:
a = ( a = (
f"({self._a})" f"({self._a})"
if isinstance(self._a, (_AddGenerator, _SubGenerator)) if isinstance(self._a, (_AddGenerator, _SubGenerator))
...@@ -338,7 +338,7 @@ class _DivGenerator(SignalGenerator): ...@@ -338,7 +338,7 @@ class _DivGenerator(SignalGenerator):
def __call__(self, time: int) -> complex: def __call__(self, time: int) -> complex:
return self._a(time) / self._b(time) return self._a(time) / self._b(time)
def __repr__(self): def __repr__(self) -> str:
a = ( a = (
f"({self._a})" f"({self._a})"
if isinstance(self._a, (_AddGenerator, _SubGenerator)) if isinstance(self._a, (_AddGenerator, _SubGenerator))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment