class TestSignalFolding: def setup_method(self, method): self.s = BaseSignal(np.zeros((2, 3, 4, 5))) self.s = self.s.transpose(signal_axes=2) def test_unfold_navigation(self): s = self.s.deepcopy() s.unfold_navigation_space() assert (s.axes_manager.navigation_shape == ( self.s.axes_manager.navigation_size, )) def test_unfold_signal(self): s = self.s.deepcopy() s.unfold_signal_space() assert (s.axes_manager.signal_shape == ( self.s.axes_manager.signal_size, )) def test_unfolded_repr(self): self.s.unfold() assert "unfolded" in repr(self.s) def test_unfold_navigation_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=True, unfold_signal=False) assert (s.axes_manager.navigation_shape == ( self.s.axes_manager.navigation_size, )) def test_unfold_signal_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=False, unfold_signal=True) assert (s.axes_manager.signal_shape == ( self.s.axes_manager.signal_size, )) def test_unfold_nothing_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=False, unfold_signal=False) assert s.data.shape == self.s.data.shape def test_unfold_full_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=True, unfold_signal=True) assert (s.axes_manager.signal_shape == ( self.s.axes_manager.signal_size, )) assert (s.axes_manager.navigation_shape == ( self.s.axes_manager.navigation_size, )) def test_unfolded_context_manager(self): s = self.s.deepcopy() with s.unfolded(): # Check that both spaces unfold as expected assert (s.axes_manager.navigation_shape == ( self.s.axes_manager.navigation_size, )) assert (s.axes_manager.signal_shape == ( self.s.axes_manager.signal_size, )) # Check that it folds back as expected assert (s.axes_manager.navigation_shape == self.s.axes_manager.navigation_shape) assert ( s.axes_manager.signal_shape == self.s.axes_manager.signal_shape) def test_unfolded_full_by_keywords(self): s = self.s.deepcopy() with s.unfolded(unfold_navigation=True, unfold_signal=True) as folded: assert folded # Check that both spaces unfold as expected assert (s.axes_manager.navigation_shape == ( self.s.axes_manager.navigation_size, )) assert (s.axes_manager.signal_shape == ( self.s.axes_manager.signal_size, )) # Check that it folds back as expected assert (s.axes_manager.navigation_shape == self.s.axes_manager.navigation_shape) assert ( s.axes_manager.signal_shape == self.s.axes_manager.signal_shape) def test_unfolded_navigation_by_keyword(self): s = self.s.deepcopy() with s.unfolded(unfold_navigation=True, unfold_signal=False) as folded: assert folded # Check that only navigation space unfolded assert (s.axes_manager.navigation_shape == ( self.s.axes_manager.navigation_size, )) assert (s.axes_manager.signal_shape == self.s.axes_manager.signal_shape) # Check that it folds back as expected assert (s.axes_manager.navigation_shape == self.s.axes_manager.navigation_shape) assert ( s.axes_manager.signal_shape == self.s.axes_manager.signal_shape) def test_unfolded_signal_by_keyword(self): s = self.s.deepcopy() with s.unfolded(unfold_navigation=False, unfold_signal=True) as folded: assert folded # Check that only signal space unfolded assert (s.axes_manager.navigation_shape == self.s.axes_manager.navigation_shape) assert (s.axes_manager.signal_shape == ( self.s.axes_manager.signal_size, )) # Check that it folds back as expected assert (s.axes_manager.navigation_shape == self.s.axes_manager.navigation_shape) assert ( s.axes_manager.signal_shape == self.s.axes_manager.signal_shape) def test_unfolded_nothin_by_keyword(self): s = self.s.deepcopy() with s.unfolded(False, False) as folded: assert not folded # Check that nothing folded assert (s.axes_manager.navigation_shape == self.s.axes_manager.navigation_shape) assert (s.axes_manager.signal_shape == self.s.axes_manager.signal_shape) # Check that it "folds back" as expected assert (s.axes_manager.navigation_shape == self.s.axes_manager.navigation_shape) assert ( s.axes_manager.signal_shape == self.s.axes_manager.signal_shape)
class TestSignalFolding: def setup_method(self, method): self.s = BaseSignal(np.zeros((2, 3, 4, 5))) self.s = self.s.transpose(signal_axes=2) def test_unfold_navigation(self): s = self.s.deepcopy() s.unfold_navigation_space() assert (s.axes_manager.navigation_shape == (self.s.axes_manager.navigation_size,)) def test_unfold_signal(self): s = self.s.deepcopy() s.unfold_signal_space() assert (s.axes_manager.signal_shape == (self.s.axes_manager.signal_size,)) def test_unfolded_repr(self): self.s.unfold() assert "unfolded" in repr(self.s) def test_unfold_navigation_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=True, unfold_signal=False) assert (s.axes_manager.navigation_shape == (self.s.axes_manager.navigation_size,)) def test_unfold_signal_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=False, unfold_signal=True) assert (s.axes_manager.signal_shape == (self.s.axes_manager.signal_size,)) def test_unfold_nothing_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=False, unfold_signal=False) assert s.data.shape == self.s.data.shape def test_unfold_full_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=True, unfold_signal=True) assert (s.axes_manager.signal_shape == (self.s.axes_manager.signal_size,)) assert (s.axes_manager.navigation_shape == (self.s.axes_manager.navigation_size,)) def test_unfolded_context_manager(self): s = self.s.deepcopy() with s.unfolded(): # Check that both spaces unfold as expected assert (s.axes_manager.navigation_shape == (self.s.axes_manager.navigation_size,)) assert (s.axes_manager.signal_shape == (self.s.axes_manager.signal_size,)) # Check that it folds back as expected assert (s.axes_manager.navigation_shape == self.s.axes_manager.navigation_shape) assert (s.axes_manager.signal_shape == self.s.axes_manager.signal_shape) def test_unfolded_full_by_keywords(self): s = self.s.deepcopy() with s.unfolded(unfold_navigation=True, unfold_signal=True) as folded: assert folded # Check that both spaces unfold as expected assert (s.axes_manager.navigation_shape == (self.s.axes_manager.navigation_size,)) assert (s.axes_manager.signal_shape == (self.s.axes_manager.signal_size,)) # Check that it folds back as expected assert (s.axes_manager.navigation_shape == self.s.axes_manager.navigation_shape) assert (s.axes_manager.signal_shape == self.s.axes_manager.signal_shape) def test_unfolded_navigation_by_keyword(self): s = self.s.deepcopy() with s.unfolded(unfold_navigation=True, unfold_signal=False) as folded: assert folded # Check that only navigation space unfolded assert (s.axes_manager.navigation_shape == (self.s.axes_manager.navigation_size,)) assert (s.axes_manager.signal_shape == self.s.axes_manager.signal_shape) # Check that it folds back as expected assert (s.axes_manager.navigation_shape == self.s.axes_manager.navigation_shape) assert (s.axes_manager.signal_shape == self.s.axes_manager.signal_shape) def test_unfolded_signal_by_keyword(self): s = self.s.deepcopy() with s.unfolded(unfold_navigation=False, unfold_signal=True) as folded: assert folded # Check that only signal space unfolded assert (s.axes_manager.navigation_shape == self.s.axes_manager.navigation_shape) assert (s.axes_manager.signal_shape == (self.s.axes_manager.signal_size,)) # Check that it folds back as expected assert (s.axes_manager.navigation_shape == self.s.axes_manager.navigation_shape) assert (s.axes_manager.signal_shape == self.s.axes_manager.signal_shape) def test_unfolded_nothin_by_keyword(self): s = self.s.deepcopy() with s.unfolded(False, False) as folded: assert not folded # Check that nothing folded assert (s.axes_manager.navigation_shape == self.s.axes_manager.navigation_shape) assert (s.axes_manager.signal_shape == self.s.axes_manager.signal_shape) # Check that it "folds back" as expected assert (s.axes_manager.navigation_shape == self.s.axes_manager.navigation_shape) assert (s.axes_manager.signal_shape == self.s.axes_manager.signal_shape)
class TestSignalFolding: def setUp(self): self.s = BaseSignal(np.zeros((2, 3, 4, 5))) self.s.axes_manager.set_signal_dimension(2) def test_unfold_navigation(self): s = self.s.deepcopy() s.unfold_navigation_space() nt.assert_equal(s.axes_manager.navigation_shape, (self.s.axes_manager.navigation_size, )) def test_unfold_signal(self): s = self.s.deepcopy() s.unfold_signal_space() nt.assert_equal(s.axes_manager.signal_shape, (self.s.axes_manager.signal_size, )) def test_unfolded_repr(self): self.s.unfold() nt.assert_true("unfolded" in repr(self.s)) def test_unfold_navigation_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=True, unfold_signal=False) nt.assert_equal(s.axes_manager.navigation_shape, (self.s.axes_manager.navigation_size, )) def test_unfold_signal_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=False, unfold_signal=True) nt.assert_equal(s.axes_manager.signal_shape, (self.s.axes_manager.signal_size, )) def test_unfold_nothing_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=False, unfold_signal=False) nt.assert_equal(s.data.shape, self.s.data.shape) def test_unfold_full_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=True, unfold_signal=True) nt.assert_equal(s.axes_manager.signal_shape, (self.s.axes_manager.signal_size, )) nt.assert_equal(s.axes_manager.navigation_shape, (self.s.axes_manager.navigation_size, )) def test_unfolded_context_manager(self): s = self.s.deepcopy() with s.unfolded(): # Check that both spaces unfold as expected nt.assert_equal(s.axes_manager.navigation_shape, (self.s.axes_manager.navigation_size, )) nt.assert_equal(s.axes_manager.signal_shape, (self.s.axes_manager.signal_size, )) # Check that it folds back as expected nt.assert_equal(s.axes_manager.navigation_shape, self.s.axes_manager.navigation_shape) nt.assert_equal(s.axes_manager.signal_shape, self.s.axes_manager.signal_shape) def test_unfolded_full_by_keywords(self): s = self.s.deepcopy() with s.unfolded(unfold_navigation=True, unfold_signal=True) as folded: nt.assert_true(folded) # Check that both spaces unfold as expected nt.assert_equal(s.axes_manager.navigation_shape, (self.s.axes_manager.navigation_size, )) nt.assert_equal(s.axes_manager.signal_shape, (self.s.axes_manager.signal_size, )) # Check that it folds back as expected nt.assert_equal(s.axes_manager.navigation_shape, self.s.axes_manager.navigation_shape) nt.assert_equal(s.axes_manager.signal_shape, self.s.axes_manager.signal_shape) def test_unfolded_navigation_by_keyword(self): s = self.s.deepcopy() with s.unfolded(unfold_navigation=True, unfold_signal=False) as folded: nt.assert_true(folded) # Check that only navigation space unfolded nt.assert_equal(s.axes_manager.navigation_shape, (self.s.axes_manager.navigation_size, )) nt.assert_equal(s.axes_manager.signal_shape, self.s.axes_manager.signal_shape) # Check that it folds back as expected nt.assert_equal(s.axes_manager.navigation_shape, self.s.axes_manager.navigation_shape) nt.assert_equal(s.axes_manager.signal_shape, self.s.axes_manager.signal_shape) def test_unfolded_signal_by_keyword(self): s = self.s.deepcopy() with s.unfolded(unfold_navigation=False, unfold_signal=True) as folded: nt.assert_true(folded) # Check that only signal space unfolded nt.assert_equal(s.axes_manager.navigation_shape, self.s.axes_manager.navigation_shape) nt.assert_equal(s.axes_manager.signal_shape, (self.s.axes_manager.signal_size, )) # Check that it folds back as expected nt.assert_equal(s.axes_manager.navigation_shape, self.s.axes_manager.navigation_shape) nt.assert_equal(s.axes_manager.signal_shape, self.s.axes_manager.signal_shape) def test_unfolded_nothin_by_keyword(self): s = self.s.deepcopy() with s.unfolded(False, False) as folded: nt.assert_false(folded) # Check that nothing folded nt.assert_equal(s.axes_manager.navigation_shape, self.s.axes_manager.navigation_shape) nt.assert_equal(s.axes_manager.signal_shape, self.s.axes_manager.signal_shape) # Check that it "folds back" as expected nt.assert_equal(s.axes_manager.navigation_shape, self.s.axes_manager.navigation_shape) nt.assert_equal(s.axes_manager.signal_shape, self.s.axes_manager.signal_shape)
class TestSignalFolding: def setUp(self): self.s = BaseSignal(np.zeros((2, 3, 4, 5))) self.s.axes_manager.set_signal_dimension(2) def test_unfold_navigation(self): s = self.s.deepcopy() s.unfold_navigation_space() nt.assert_equal(s.axes_manager.navigation_shape, (self.s.axes_manager.navigation_size,)) def test_unfold_signal(self): s = self.s.deepcopy() s.unfold_signal_space() nt.assert_equal(s.axes_manager.signal_shape, (self.s.axes_manager.signal_size,)) def test_unfolded_repr(self): self.s.unfold() nt.assert_true("unfolded" in repr(self.s)) def test_unfold_navigation_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=True, unfold_signal=False) nt.assert_equal(s.axes_manager.navigation_shape, (self.s.axes_manager.navigation_size,)) def test_unfold_signal_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=False, unfold_signal=True) nt.assert_equal(s.axes_manager.signal_shape, (self.s.axes_manager.signal_size,)) def test_unfold_nothing_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=False, unfold_signal=False) nt.assert_equal(s.data.shape, self.s.data.shape) def test_unfold_full_by_keyword(self): s = self.s.deepcopy() s.unfold(unfold_navigation=True, unfold_signal=True) nt.assert_equal(s.axes_manager.signal_shape, (self.s.axes_manager.signal_size,)) nt.assert_equal(s.axes_manager.navigation_shape, (self.s.axes_manager.navigation_size,)) def test_unfolded_context_manager(self): s = self.s.deepcopy() with s.unfolded(): # Check that both spaces unfold as expected nt.assert_equal(s.axes_manager.navigation_shape, (self.s.axes_manager.navigation_size,)) nt.assert_equal(s.axes_manager.signal_shape, (self.s.axes_manager.signal_size,)) # Check that it folds back as expected nt.assert_equal(s.axes_manager.navigation_shape, self.s.axes_manager.navigation_shape) nt.assert_equal(s.axes_manager.signal_shape, self.s.axes_manager.signal_shape) def test_unfolded_full_by_keywords(self): s = self.s.deepcopy() with s.unfolded(unfold_navigation=True, unfold_signal=True) as folded: nt.assert_true(folded) # Check that both spaces unfold as expected nt.assert_equal(s.axes_manager.navigation_shape, (self.s.axes_manager.navigation_size,)) nt.assert_equal(s.axes_manager.signal_shape, (self.s.axes_manager.signal_size,)) # Check that it folds back as expected nt.assert_equal(s.axes_manager.navigation_shape, self.s.axes_manager.navigation_shape) nt.assert_equal(s.axes_manager.signal_shape, self.s.axes_manager.signal_shape) def test_unfolded_navigation_by_keyword(self): s = self.s.deepcopy() with s.unfolded(unfold_navigation=True, unfold_signal=False) as folded: nt.assert_true(folded) # Check that only navigation space unfolded nt.assert_equal(s.axes_manager.navigation_shape, (self.s.axes_manager.navigation_size,)) nt.assert_equal(s.axes_manager.signal_shape, self.s.axes_manager.signal_shape) # Check that it folds back as expected nt.assert_equal(s.axes_manager.navigation_shape, self.s.axes_manager.navigation_shape) nt.assert_equal(s.axes_manager.signal_shape, self.s.axes_manager.signal_shape) def test_unfolded_signal_by_keyword(self): s = self.s.deepcopy() with s.unfolded(unfold_navigation=False, unfold_signal=True) as folded: nt.assert_true(folded) # Check that only signal space unfolded nt.assert_equal(s.axes_manager.navigation_shape, self.s.axes_manager.navigation_shape) nt.assert_equal(s.axes_manager.signal_shape, (self.s.axes_manager.signal_size,)) # Check that it folds back as expected nt.assert_equal(s.axes_manager.navigation_shape, self.s.axes_manager.navigation_shape) nt.assert_equal(s.axes_manager.signal_shape, self.s.axes_manager.signal_shape) def test_unfolded_nothin_by_keyword(self): s = self.s.deepcopy() with s.unfolded(False, False) as folded: nt.assert_false(folded) # Check that nothing folded nt.assert_equal(s.axes_manager.navigation_shape, self.s.axes_manager.navigation_shape) nt.assert_equal(s.axes_manager.signal_shape, self.s.axes_manager.signal_shape) # Check that it "folds back" as expected nt.assert_equal(s.axes_manager.navigation_shape, self.s.axes_manager.navigation_shape) nt.assert_equal(s.axes_manager.signal_shape, self.s.axes_manager.signal_shape)