How to use test_write_csv_files method in locust

Best Python code snippet using locust

test_covid19_processing.py

Source:test_covid19_processing.py

...85    processing = Covid19Processing('test_dir')86    with patch('os.mkdir') as mock_mkdir:87        processing.create_out_dir()88        mock_mkdir.assert_called_once_with('test_dir')89def test_write_csv_files():90    """Tests write_csv_files"""91    processing = Covid19Processing('test_dir')92    with patch('pandas.DataFrame.to_csv') as mock_to_csv:93        processing.cases_data = pd.DataFrame(['test data'])94        processing.deaths_data = pd.DataFrame(['test data'])95        processing.recovered_data = pd.DataFrame(['test data'])96        processing.write_csv_files()97        mock_to_csv.assert_any_call('test_dir/cases.csv')98        mock_to_csv.assert_any_call('test_dir/deaths.csv')99        mock_to_csv.assert_any_call('test_dir/recovered.csv')100def test_remove_unrecongnised_column():101    """Tests remove_unrecognised_column"""102    processing = Covid19Processing('test_dir')103    processing.cases_data = pd.DataFrame(...

test_pyvolcans.py

Source:test_pyvolcans.py

...54        assert top1_analogue['name'] == expected55@pytest.mark.parametrize("input_args,expected",56                         [("Alutu -w", "222040"),57                          ("Vulcano -w", "263310")])58def test_write_csv_files(input_args, expected, tmp_path):59    # Arrange / Action60    subprocess.run(['pyvolcans', '--verbose', *input_args.split()],61                    cwd=tmp_path)62    fname = list(Path(tmp_path).glob("*.csv"))[0]63    data = pd.read_csv(fname)64    top1_analogue = data.iloc[0, ]65    # Assert66    assert top1_analogue['smithsonian_id'] == int(expected)67@pytest.mark.parametrize("input_args,expected",68                         [("Fuego -G 99", "PyVOLCANS: Sum of weights"),69                          ("Fuego -Sz -1", "PyVOLCANS: Some criterion weights"),70                          ("Fuigu", "Fuigu not found!"),71                          ("342090111", "Volcano number does not exist"),72                          ("-iurwhfgl", "arguments are required: volcano"),...

TestConverterToCsv.py

Source:TestConverterToCsv.py

...25    def test_get_tables_contents_from_url(self):26        urlsTab = "https://en.wikipedia.org/wiki/Comparison_between_Esperanto_and_Ido"27        tabs = md.ConverterToCsv.get_tables_contents_from_url(urlsTab)28        self.assertEqual(8, len(tabs))29    def test_write_csv_files(self):30        # path31        monRepertoire_output = '../output'32        #  Get the list of all files and directories33        #  in the root directory 'path'34        for folder in listdir(monRepertoire_output):35            self.assertTrue(isfile(join(monRepertoire_output, folder)))36    def test_count_csv_files(self):37        path_output = '../output'38        nbFiles = 039        for folder in listdir(path_output):40            if isfile(join(path_output, folder)):41                nbFiles = nbFiles + 142        self.assertEqual(nbFiles, md.ConverterToCsv.count_csv_files(path_output))43if __name__ == '__main__':...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.