Java-CSV文件读取

复制代码
 1 import java.io.BufferedReader;
 2 import java.io.FileInputStream;
 3 import java.io.IOException;
 4 import java.io.InputStreamReader;
 5 import java.util.ArrayList;
 6 import java.util.List;
 7 import org.testng.annotations.DataProvider;
 8 import org.testng.annotations.Test;
 9 
10 /**
11  * 读取csv文件
12  */
13 public class File_CSV {
14     // 测试案例绑定DataProvider后自动循环执行
15     @Test(dataProvider="getData")
16     public void testSearch(String p1, String p2) {
17         System.out.println(p1 + " " + p2);
18     }
19         
20     // 定义数据驱动
21     @DataProvider(name = "getData")
22     public static Object[][] getData() throws IOException {
23         return getTestData(".\\testData\\testDate.csv");
24     }
25 
26     // 读取CSV文件的静态方法,使用CSV文件的绝对文件路径作为函数参数
27     public static Object[][] getTestData(String fileName) throws IOException {
28         List<String[]> records = new ArrayList<String[]>();
29         String record;
30         // 设定UTF-8字符集,使用带缓冲区的字符输入流BufferedReader读取文件内容
31         BufferedReader file = new BufferedReader(new InputStreamReader(new FileInputStream(fileName), "UTF-8"));
32         // file.readLine(); //跳过表头所在的行
33 
34         // 遍历数据行并存储在名为records的ArrayList中,每一行records中存储的对象为一个String数组
35         while ((record = file.readLine()) != null) {
36             String fields[] = record.split(",");
37             records.add(fields);
38         }
39         // 关闭文件
40         file.close();
41         // 将存储测试数据的List转换为一个Object的二维数组
42         Object[][] results = new Object[records.size()][];
43         // 设置二位数组每行的值,每行是一个Object对象
44         for (int i = 0; i < records.size(); i++) {
45             results[i] = (Object[]) records.get(i);
46         }
47         return results;
48     }
49 }
复制代码

testDate.csv文件内容:

  

 

posted @   为你编程  阅读(13668)  评论(0编辑  收藏  举报
(评论功能已被禁用)
编辑推荐:
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
阅读排行:
· Manus爆火,是硬核还是营销?
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
点击右上角即可分享
微信分享提示