예제 #1
0
 def test_read_csv(self):
     path = '{}/user/xpatterns/files/test-frame.csv'.format(hdfs_prefix)
     res = XFrame.read_csv(path)
     self.assertEqualLen(3, res)
     self.assertListEqual(['id', 'val'], res.column_names())
     self.assertListEqual([int, str], res.column_types())
     self.assertDictEqual({'id': 1, 'val': 'a'}, res[0])
     self.assertDictEqual({'id': 2, 'val': 'b'}, res[1])
     self.assertDictEqual({'id': 3, 'val': 'c'}, res[2])