test_utils.test_csv_to_pd()

Complexity

Conditions

3

Size

Total Lines	43
Code Lines	39

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

import tempfile
def test_csv_to_pd():
    tf = tempfile.NamedTemporaryFile(suffix=".csv")
    fid = open(tf.name, 'w')
    fid.write("This file won't work. \n This file won't work. \n This file won't work.")
    fid.close()
    
    stdout_ = sys.stdout 
    stream = StringIO()
    sys.stdout = stream
    test = ut.csv_to_pd(tf.name)
    sys.stdout = stdout_
    variable = stream.getvalue()
    test_string = 'No data in csv file.\n'
    assert(variable==test_string)
    
    d = {'Track_ID': [],
         'Spot_ID': [],
         'Frame': [],
         'X': [],
         'Y': [],
         'Quality': [],
         'SN_Ratio': [],
         'Mean_Intensity': []}
    cols = ['Track_ID', 'Spot_ID', 'Frame', 'X', 'Y', 'Quality', 'SN_Ratio', 'Mean_Intensity']
    data = pd.DataFrame(data=d, index=[])
    data = data[cols]
    data = data.astype('float64')
    pdt.assert_frame_equal(test, data)
    
    tf = tempfile.NamedTemporaryFile(suffix=".csv")
    fid = open(tf.name, 'w')
    fid.write('Found 0 tracks.\nData starts here.\nTrack_ID,Spot_ID,Frame,X,Y,Quality,SN_Ratio,Mean_Intensity\n')
    fid.close()
    
    stdout_ = sys.stdout 
    stream = StringIO()
    sys.stdout = stream
    test = ut.csv_to_pd(tf.name)
    sys.stdout = stdout_
    variable = stream.getvalue()
    test_string = ''
    assert(variable==test_string)
    pdt.assert_frame_equal(test, data)