fix broken test

KlugerLab · Aug 6, 2024 · f57bb0e · f57bb0e
1 parent 7697a6f
commit f57bb0e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_dyson_equalizer.py b/tests/test_dyson_equalizer.py
@@ -3,19 +3,19 @@
 import numpy as np
 
 from dyson_equalizer.dyson_equalizer import DysonEqualizer
-from dyson_equalizer.examples import generate_X_with_correlated_noise
+from dyson_equalizer.examples import generate_Y_with_correlated_noise
 
 
 class DysonEqualizerTestCase(unittest.TestCase):
     def test_fig_2(self):
-        Y = generate_X_with_correlated_noise()
+        Y = generate_Y_with_correlated_noise()
         de = DysonEqualizer(Y)
         de.compute()
 
         self.assertEqual(20, de.r_hat)
 
     def test_transpose(self):
-        Y = generate_X_with_correlated_noise()
+        Y = generate_Y_with_correlated_noise()
 
         de = DysonEqualizer(Y)
         de.compute()