def test_read_csv_types(self): path = str(test.get_data_path("vehicles.csv")) types = {"hwy": int, "cty": int} data = ListOfDicts.read_csv(path, types=types) assert isinstance(data[0].hwy, int) assert isinstance(data[0].cty, int) assert isinstance(data[100].hwy, int) assert isinstance(data[100].cty, int)
def test_read_json_types(self): path = str(test.get_data_path("downloads.json")) # datetime.date.fromisoformat requires Python >= 3.7. date = lambda x: datetime.date(*map(int, x.split("-"))) types = {"date": date, "downloads": int} data = ListOfDicts.read_json(path, types=types) assert isinstance(data[0].date, datetime.date) assert isinstance(data[0].downloads, int) assert isinstance(data[100].date, datetime.date) assert isinstance(data[100].downloads, int)
def test_read_json_keys(self): path = str(test.get_data_path("downloads.json")) data = ListOfDicts.read_json(path, keys=["date", "downloads"]) assert list(data[0].keys()) == ["date", "downloads"] assert list(data[100].keys()) == ["date", "downloads"]
def test_read_json(self): path = str(test.get_data_path("downloads.json")) data = ListOfDicts.read_json(path) assert len(data) == 905 assert all(len(x) == 3 for x in data)
def test_read_csv_path(self): ListOfDicts.read_csv(test.get_data_path("vehicles.csv"))
def test_read_csv_keys(self): path = str(test.get_data_path("vehicles.csv")) data = ListOfDicts.read_csv(path, keys=["make", "model"]) assert list(data[0].keys()) == ["make", "model"] assert list(data[100].keys()) == ["make", "model"]
def test_read_csv(self): path = str(test.get_data_path("vehicles.csv")) data = ListOfDicts.read_csv(path) assert len(data) == 33442 assert all(len(x) == 12 for x in data)
def test_read_csv_path(self): DataFrame.read_csv(test.get_data_path("vehicles.csv"))
def test_read_dtypes(self): path = str(test.get_data_path(self.path)) dtypes = {"neighbourhood": object, "neighbourhood_group": object} data = GeoJSON.read(path, dtypes=dtypes) assert data.neighbourhood.is_object() assert data.neighbourhood_group.is_object()
def test_read_columns(self): path = str(test.get_data_path(self.path)) data = GeoJSON.read(path, columns=["neighbourhood"]) assert data.colnames == ["neighbourhood", "geometry"]
def test_read(self): path = str(test.get_data_path(self.path)) data = GeoJSON.read(path) assert data.nrow == 233 assert data.ncol == 3
def test_read_json_path(self): DataFrame.read_json(test.get_data_path("vehicles.json"))
def test_read_json_dtypes(self): path = str(test.get_data_path("vehicles.json")) dtypes = {"make": object, "model": object} data = DataFrame.read_json(path, dtypes=dtypes) assert data.make.is_object() assert data.model.is_object()
def test_read_json_columns(self): path = str(test.get_data_path("vehicles.json")) data = DataFrame.read_json(path, columns=["make", "model"]) assert data.colnames == ["make", "model"]
def test_read_json(self): path = str(test.get_data_path("downloads.json")) data = DataFrame.read_json(path) assert data.nrow == 905 assert data.ncol == 3
def test_read_json_path(self): ListOfDicts.read_csv(test.get_data_path("vehicles.json"))
def test_read_path(self): GeoJSON.read(test.get_data_path(self.path))
def test_read_csv(self): path = str(test.get_data_path("vehicles.csv")) data = DataFrame.read_csv(path) assert data.nrow == 33442 assert data.ncol == 12