def test_jsonl(self): result_0 = seq.jsonl('functional/test/data/test.jsonl').to_list() expect_0 = [[1, 2, 3], {'a': 1, 'b': 2, 'c': 3}] self.assertEqual(expect_0, result_0) result_1 = seq.jsonl(['[1, 2, 3]', '[4, 5, 6]']) expect_1 = [[1, 2, 3], [4, 5, 6]] self.assertEqual(expect_1, result_1)
def test_to_jsonl(self): tmp_path = 'functional/test/data/tmp/output.txt' elements = [{'a': 1, 'b': 2}, {'c': 3}, {'d': 4}] sequence = seq(elements) sequence.to_jsonl(tmp_path) result = seq.jsonl(tmp_path).to_list() self.assertEqual(elements, result)
def test_seq_jsonl(self): f = _make_str_file(u''' [11, 12, 13] [21, 22, 23] [31, 32, 33] ''') res = seq.jsonl(f) assert res == [[11, 12, 13], [21, 22, 23], [31, 32, 33]]