rangecut implements wavenumbers_ when cut

Author: paucablop

chemotools/feature_selection/_range_cut.py

@@ -78,6 +78,7 @@ def fit(self, X: np.ndarray, y=None) -> "RangeCut":
         else:
             self.start_index_ = self._find_index(self.start)
             self.end_index_ = self._find_index(self.end)
+            self.wavenumbers_ = self.wavenumbers[self.start_index_ : self.end_index_]
 
         return self
 

tests/test_functionality.py

@@ -622,6 +622,7 @@ def test_range_cut_by_wavenumber_with_list():
 
     # Assert
     assert np.allclose(spectrum_corrected[0], spectrum[0][1:7], atol=1e-8)
+    assert range_cut.wavenumbers_ == [2, 3, 4, 5, 6, 7]
 
 
 def test_range_cut_by_wavenumber_with_dataframe():