check bounded spectrum on combinatorial Laplacian

epfl-lts2 · Dec 4, 2018 · 58421d5 · 58421d5
1 parent 4227b49
commit 58421d5
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pygsp/graphs/fourier.py b/pygsp/graphs/fourier.py
@@ -181,9 +181,11 @@ def compute_fourier_basis(self, n_eigenvectors=None, recompute=False):
         assert -1e-12 < self._e[0] < 1e-12
         self._e[0] = 0
 
-        if self.lap_type == 'normalized':
-            # Spectrum bounded by [0, 2].
-            assert self._e[-1] <= 2
+        # Bounded spectrum.
+        if self.lap_type == 'combinatorial':
+            assert self._e[-1] <= 2 * np.max(self.dw) + 1e-12
+        elif self.lap_type == 'normalized':
+            assert self._e[-1] <= 2 + 1e-12
 
         assert np.max(self._e) == self._e[-1]
         if n_eigenvectors == self.N: