def test_validate_dataframes_valid(self): df1 = { 'Unknown': { 'sink1': 0.25, 'sink2': 0.25 }, 'Source1': { 'sink1': 0.50, 'sink2': 0.25 }, 'Source2': { 'sink1': 0.25, 'sink2': 0.25 } } df1 = pd.DataFrame(df1) df2 = { 'Unknown': { 'sink1': 0.1, 'sink2': 0.25 }, 'Source1': { 'sink1': 0.1, 'sink2': 0.25 }, 'Source2': { 'sink1': 0.8, 'sink2': 0.25 } } df2 = pd.DataFrame(df2) _validate_dataframes(df1, df2)
def test_validate_dataframes_valid(self): df1 = {'Unknown': {'sink1': 0.25, 'sink2': 0.25}, 'Source1': {'sink1': 0.50, 'sink2': 0.25}, 'Source2': {'sink1': 0.25, 'sink2': 0.25}} df1 = pd.DataFrame(df1) df2 = {'Unknown': {'sink1': 0.1, 'sink2': 0.25}, 'Source1': {'sink1': 0.1, 'sink2': 0.25}, 'Source2': {'sink1': 0.8, 'sink2': 0.25}} df2 = pd.DataFrame(df2) _validate_dataframes(df1, df2)
def test_validate_dataframes_invalid_sources(self): df1 = {'Hello': {'sink1': 0.25, 'sink2': 0.25}, 'Source1': {'sink1': 0.50, 'sink2': 0.25}, 'Source2': {'sink1': 0.25, 'sink2': 0.25}} df1 = pd.DataFrame(df1) df2 = {'Unknown': {'sink1': 0.1, 'sink2': 0.25}, 'Source1': {'sink1': 0.1, 'sink2': 0.25}, 'Source2': {'sink1': 0.8, 'sink2': 0.25}} df2 = pd.DataFrame(df2) with self.assertRaises(ValueError): _validate_dataframes(df1, df2) df1 = {'Source1': {'sink1': 0.50, 'sink2': 0.25}, 'Source2': {'sink1': 0.50, 'sink2': 0.25}} df1 = pd.DataFrame(df1) df2 = {'Unknown': {'sink1': 0.1, 'sink2': 0.25}, 'Source1': {'sink1': 0.1, 'sink2': 0.25}, 'Source2': {'sink1': 0.8, 'sink2': 0.25}} df2 = pd.DataFrame(df2) with self.assertRaises(ValueError): _validate_dataframes(df1, df2)
def test_validate_dataframes_invalid_sources(self): df1 = { 'Hello': { 'sink1': 0.25, 'sink2': 0.25 }, 'Source1': { 'sink1': 0.50, 'sink2': 0.25 }, 'Source2': { 'sink1': 0.25, 'sink2': 0.25 } } df1 = pd.DataFrame(df1) df2 = { 'Unknown': { 'sink1': 0.1, 'sink2': 0.25 }, 'Source1': { 'sink1': 0.1, 'sink2': 0.25 }, 'Source2': { 'sink1': 0.8, 'sink2': 0.25 } } df2 = pd.DataFrame(df2) with self.assertRaises(ValueError): _validate_dataframes(df1, df2) df1 = { 'Source1': { 'sink1': 0.50, 'sink2': 0.25 }, 'Source2': { 'sink1': 0.50, 'sink2': 0.25 } } df1 = pd.DataFrame(df1) df2 = { 'Unknown': { 'sink1': 0.1, 'sink2': 0.25 }, 'Source1': { 'sink1': 0.1, 'sink2': 0.25 }, 'Source2': { 'sink1': 0.8, 'sink2': 0.25 } } df2 = pd.DataFrame(df2) with self.assertRaises(ValueError): _validate_dataframes(df1, df2)