|
8 | 8 | from unittest import TestCase |
9 | 9 |
|
10 | 10 | import numpy |
11 | | -import numpy.compat |
| 11 | +import numpy.fft |
12 | 12 | import numpy.linalg as linalg |
13 | 13 | from rich.progress import Progress |
14 | 14 |
|
@@ -56,17 +56,17 @@ def test_lazy_module(self): |
56 | 56 |
|
57 | 57 | # 1.2 getattr as LazyAttr |
58 | 58 | self.assertIsInstance(lazy_numpy.linalg, LazyAttr) |
59 | | - self.assertIsInstance(lazy_numpy.compat, LazyAttr) |
| 59 | + self.assertIsInstance(lazy_numpy.fft, LazyAttr) |
60 | 60 |
|
61 | | - # 1.3 Build module from LazyObject. amp and functional can be accessed |
| 61 | + # 1.3 Build module from LazyObject. linalg and fft can be accessed |
62 | 62 | imported_numpy = lazy_numpy.build() |
63 | 63 | self.assertIs(imported_numpy.linalg, linalg) |
64 | | - self.assertIs(imported_numpy.compat, numpy.compat) |
| 64 | + self.assertIs(imported_numpy.fft, numpy.fft) |
65 | 65 |
|
66 | 66 | # 1.4.1 Build module from LazyAttr |
67 | 67 | imported_linalg = lazy_numpy.linalg.build() |
68 | | - imported_compat = lazy_numpy.compat.build() |
69 | | - self.assertIs(imported_compat, numpy.compat) |
| 68 | + imported_fft = lazy_numpy.fft.build() |
| 69 | + self.assertIs(imported_fft, numpy.fft) |
70 | 70 | self.assertIs(imported_linalg, linalg) |
71 | 71 |
|
72 | 72 | # 1.4.2 build class method from LazyAttr |
|
0 commit comments