def test_cvs2volstr(): """Test that we can produce strings from dataframes""" data = csv2ofmvol.read_pdm_csv_files([PRODDATA_A3, PRODDATA_A4]) volstr = csv2ofmvol.df2vol(data) assert isinstance(volstr, str) dupdata = csv2ofmvol.read_pdm_csv_files([PRODDATA_A3, PRODDATA_A4, PRODDATA_A3]) assert len(dupdata) == len(data)
def test_read_pdf_csv_files(): """Test parsing of CSV or dataframes""" # pylint: disable=no-value-for-parameter with pytest.raises(TypeError): csv2ofmvol.read_pdm_csv_files() with pytest.raises(IOError): csv2ofmvol.read_pdm_csv_files(["foobar"]) with pytest.raises(IOError): csv2ofmvol.read_pdm_csv_files("foobar") processeddata = csv2ofmvol.read_pdm_csv_files(PRODDATA_A3) assert processeddata.index.names[0] == "WELL" assert processeddata.index.names[1] == "DATE" assert processeddata.columns == ["WOPR"] # DATE and WELL is index assert len(processeddata) == len(PRODDATA_A3) processeddata = csv2ofmvol.read_pdm_csv_files([PRODDATA_A3, PRODDATA_A4]) assert len(processeddata) == len(PRODDATA_A3) + len(PRODDATA_A4) with pytest.raises(ValueError): csv2ofmvol.read_pdm_csv_files(pd.DataFrame())