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