Esempio n. 1
0
def test_multiply_vector():
    df = pd.read_csv(get_path('test_data.csv'), header=[0, 1])
    result = multiply_vector.multiply_vector(df, get_path('vector.csv'))
    expected = [1646.12, 1548.12, 1656.12, 1656.12, 1732.12]
    np.testing.assert_array_almost_equal(expected,
                                         np.array(result['wvl'].iloc[0, 0:5]))

    result = multiply_vector.multiply_vector(df, get_path('bad_vector.csv'))
    assert result == 0
 def multiply_vector(self, vectorfile):
     self.df = multiply_vector.multiply_vector(self.df,
                                               vectorfile=vectorfile)