def test_concat_empty_transform_zero_transform_trivial(self): # Build the transforms A = Transform(size_in=2, transform=0, size_out=2) B = Transform(size_in=2, transform=1, size_out=2) # Check that the test is correct expected = np.dot(B.full_transform(False, False), A.full_transform(False, False)) assert not np.any(expected), "Test broken" # Combine the transforms, this should return None to indicate that the # transform is empty. assert A.concat(B) is None
def test_concat_empty_transform_mismatched_slicing(self): # Build the transforms A = Transform(size_in=1, transform=1, size_out=2, slice_out=[0]) B = Transform(size_in=2, slice_in=[1], transform=1, size_out=1) # Check that the test is correct expected = np.dot(B.full_transform(False, False), A.full_transform(False, False)) assert not np.any(expected), "Test broken" # Combine the transforms, this should return None to indicate that the # transform is empty. assert A.concat(B) is None
def test_concat_empty_transform_zero_transform_trivial(self): # Build the transforms A = Transform(size_in=2, transform=0, size_out=2) B = Transform(size_in=2, transform=1, size_out=2) # Check that the test is correct expected = np.dot(B.full_transform(False, False), A.full_transform(False, False)) assert not np.any(expected), "Test broken" # Combine the transforms, this should return None to indicate that the # transform is empty. assert A.concat(B) is None
def test_concat_empty_transform_mismatched_slicing(self): # Build the transforms A = Transform(size_in=1, transform=1, size_out=2, slice_out=[0]) B = Transform(size_in=2, slice_in=[1], transform=1, size_out=1) # Check that the test is correct expected = np.dot(B.full_transform(False, False), A.full_transform(False, False)) assert not np.any(expected), "Test broken" # Combine the transforms, this should return None to indicate that the # transform is empty. assert A.concat(B) is None
def test_concat(self, a_params, b_params): # Build the transforms A = Transform(size_in=8, size_out=8, **a_params) B = Transform(size_in=8, size_out=4, **b_params) # Compute the expected combined transform expected = np.dot(B.full_transform(False, False), A.full_transform(False, False)) # Combine the transforms C = A.concat(B) # Test assert np.array_equal(expected, C.full_transform(False, False))
def test_concat(self, a_params, b_params): # Build the transforms A = Transform(size_in=8, size_out=8, **a_params) B = Transform(size_in=8, size_out=4, **b_params) # Compute the expected combined transform expected = np.dot(B.full_transform(False, False), A.full_transform(False, False)) # Combine the transforms C = A.concat(B) # Test assert np.array_equal(expected, C.full_transform(False, False))