def test_head(self): bf = BatFrame(DatasetTest.dataset_url) head = bf.head().toPandas() self.assertTupleEqual(head.shape, (5, 8))