In [6]:

    

import background_removal
import histogram_equalization

import numpy as np


    

    




---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-6-5f4002e88b8f> in <module>()
----> 1 import background_removal
      2 import histogram_equalization
      3 
      4 import numpy as np

ImportError: No module named 'background_removal'

In [5]:

    

def test_background_removal():
    """
    This function tests the background removal function
    """
    testdata = np.genfromtxt('../Data/UnbackgroundedTXT/5umUniformNetwork')
    bkgd = background_removal(testdata)
    
    assert bkgd != None, 'Backgrounding has deleted datafile'
    return


    

In [7]:

    

def test_histogram_equalization():
    """
    This function tests the histogram equalization function
    """
    
    testdata = np.genfromtxt('../Data/UnbackgroundedTXT/5umUniformNetwork')
    
    equ = histogram_equalization(testdata)
    
    assert equ != None, 'Equalization has deleted datafile'
    return


    

In [ ]: