def test_reader_dataframe(reader):
dataframe = reader.dataframe
assert isinstance(dataframe, pd.DataFrame)
assert dataframe.shape == (1603, 8)
dtypes = dataframe.dtypes
names = list(dataframe)
assert names[0] == "timestamp"
assert types.is_datetime64tz_dtype(dtypes[0])
assert names[1] == "position"
assert types.is_object_dtype(dtypes[1]) # is GeoSeries?!?
assert names[2] == "distance"
assert types.is_float_dtype(dtypes[2])
assert names[3] == "speed"
assert types.is_float_dtype(dtypes[3])
assert names[4] == "altitude"
assert types.is_float_dtype(dtypes[4])
assert names[5] == "heart_rate"
assert types.is_float_dtype(dtypes[5])
assert names[6] == "power"
assert types.is_unsigned_integer_dtype(dtypes[6])
def contains_op(cls, series: pd.Series) -> bool:
return pdt.is_unsigned_integer_dtype(series)
def count_contains(series: pd.Series, state: dict) -> bool:
return pdt.is_unsigned_integer_dtype(series)
def make_normalized_dataframe_assertions(self, df):
self.assertEqual(df.index.name, "rsid")
self.assertTrue(is_object_dtype(df.index.dtype))
self.assertTrue(is_object_dtype(df.chrom.dtype))
self.assertTrue(is_unsigned_integer_dtype(df.pos.dtype))
self.assertTrue(is_object_dtype(df.genotype.dtype))
