View GitHub Repository torchio

Schedule Inspection
Subscribe
Passed
Push — master ( b9ac52...6aebda )
by Fernando
10:37 queued 20s
created

TestRemapLabels.test_remap()   A

↳ Parent: tests.transforms.label.test_remap_labels

Complexity

Conditions 1

Size

Total Lines 11
Code Lines 9

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 1
eloc 9
nop 1
dl 0
loc 11
rs 9.95
c 0
b 0
f 0
1 
from torchio.transforms import RemapLabels
2 
from ...utils import TorchioTestCase
3 

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    4
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    5
                

                                    
                                                    
                
                
                
class TestRemapLabels(TorchioTestCase):

            


            
                                                                                                            
                                                                
            
                                    
            
            

                
                    6
                

                                    
                                                    
                
                
                
    """Tests for `RemapLabels`."""

            


            
                                                        
            
                                    
            
            

                
                    7
                

                                    
                                                    
                
                
                
    def test_remap(self):

            


            
                                                        
            
                                    
            
            

                
                    8
                

                                    
                                                    
                
                
                
        remapping = {1: 2, 2: 1, 5: 10, 6: 11}

            


            
                                                        
            
                                    
            
            

                
                    9
                

                                    
                                                    
                
                
                
        remap_labels = RemapLabels(remapping=remapping)

            


            
                                                        
            
                                    
            
            

                
                    10
                

                                    
                                                    
                
                
                

            


            
                                                        
            
                                    
            
            

                
                    11
                

                                    
                                                    
                
                
                
        subject = self.get_subject_with_labels(labels=remapping.keys())

            


            
                                                        
            
                                    
            
            

                
                    12
                

                                    
                                                    
                
                
                
        transformed = remap_labels(subject)

            


            
                                                        
            
                                    
            
            

                
                    13
                

                                    
                                                    
                
                
                
        inverse_transformed = transformed.apply_inverse_transform()

            


            
                                                        
            
                                    
            
            

                
                    14
                

                                    
                                                    
                
                
                

            


            
                                                        
            
                                    
            
            

                
                    15
                

                                    
                                                    
                
                
                
        self.assertEqual(self.get_unique_labels(subject.label), set(remapping.keys()))

            


            
                                                        
            
                                    
            
            

                
                    16
                

                                    
                                                    
                
                
                
        self.assertEqual(self.get_unique_labels(transformed.label), set(remapping.values()))

            


            
                                                        
            
                                    
            
            

                
                    17
                

                                    
                                                    
                
                
                
        self.assertEqual(self.get_unique_labels(inverse_transformed.label), set(remapping.keys()))

            


            
                                                        
            
                                    
            
            

                
                    18