コード例 #1
0
def test_gzopen_json():
    with tmpfile('.json.gz') as filename:
        with gzip.open(filename, 'w') as f:
            f.write('[[1, 1], [2, 2]]')

        # Not a valid JSON file
        assert raises(Exception,
                      lambda: list(JSON(filename, schema='2 * int')))

        dd = JSON(filename, schema='2 * int', open=gzip.open)

        assert list(dd) == [[1, 1], [2, 2]]
コード例 #2
0
def test_gzopen_json():
    with tmpfile('.json.gz') as filename:
        f = gzip.open(filename, 'wt')
        f.write('[[1, 1], [2, 2]]')
        f.close()

        # Not a valid JSON file
        assert raises(Exception,
                      lambda: list(JSON(filename, schema='2 * int')))

        dd = JSON(filename, schema='2 * int', open=gzip.open)

        assert tuplify(list(dd)) == ((1, 1), (2, 2))
コード例 #3
0
 def test_iter(self):
     dd = JSON(self.json_file, dshape=json_dshape)
     # This equality does not work yet
     # self.assertEqual(dd.dshape, datashape.dshape(
     #     'Var, %s' % json_schema))
     print(list(dd))
     self.assertEqual(list(dd), [1, 2, 3, 4, 5])
コード例 #4
0
ファイル: test_integrative.py プロジェクト: vitan/blaze
    def test_gzip_json_files(self):
        with filetexts(texts, open=gzip.open) as filenames:
            descriptors = [JSON(fn, dshape=schema, open=gzip.open)
                           for fn in sorted(filenames)]
            dd = Stack(descriptors)

            self.assertEqual(sorted(dd), sorted(tuples))

            self.assertEqual(dd.schema, dshape(schema))
コード例 #5
0
ファイル: test_mongo.py プロジェクト: vitan/blaze
def test_jsonarray_into_mongodb(empty_collec):

    filename = tempfile.mktemp(".json")
    with open(filename, "w") as f:
        json.dump(data, f)

    dd = JSON(filename,
              schema="3 * { id : string, name : string, "
              "posts : var * { content : string, title : string },"
              " tv_show : string }")
    coll = empty_collec
    into(coll, dd, json_array=True)

    mongo_data = list(coll.find({}, {'_id': 0}))

    assert mongo_data[0] == data[0]
コード例 #6
0
ファイル: test_mongo.py プロジェクト: vitan/blaze
def test_json_into_mongodb(empty_collec):

    with filetext(json.dumps(les_mis_data)) as filename:

        dd = JSON(filename)
        coll = empty_collec
        into(coll, dd)

        mongo_data = list(coll.find())

        last = mongo_data[0]['nodes'][-1]
        first = mongo_data[0]['nodes'][0]

        first = (first['group'], first['name'])
        last = (last['group'], last['name'])

        assert dd.as_py()[1][-1] == last
        assert dd.as_py()[1][0] == first
コード例 #7
0
 def test_basic_object_type(self):
     dd = JSON(self.json_file, dshape=json_dshape)
     self.assertEqual(list(dd), [1, 2, 3, 4, 5])
コード例 #8
0
 def test_raise_error_on_non_existent_file(self):
     self.assertRaises(ValueError,
                 lambda: JSON('does-not-exist23424.josn', 'r'))
コード例 #9
0
 def test_as_py(self):
     dd = JSON(self.filename, 'r', dshape=self.dshape)
     self.assertEqual(dd.as_py(), self.data)
コード例 #10
0
 def test_basic(self):
     dd = JSON(self.filename, 'r', dshape=self.dshape)
     self.assertEqual(list(dd),
                      [nd.as_py(nd.parse_json(self.dshape,
                          json.dumps(self.data)))])
コード例 #11
0
ファイル: test_json.py プロジェクト: vitan/blaze
 def test_discovery(self):
     dd = JSON(self.filename, 'r')
     s = str(dd.dshape)
     for word in ['Thumbnail', 'string', 'int', 'images', 'type']:
         assert word in s
コード例 #12
0
ファイル: test_json.py プロジェクト: vitan/blaze
 def test_as_py(self):
     dd = JSON(self.filename, 'r', dshape=self.dshape)
     self.assertEqual(tuplify(dd.as_py()), self.ordered)
コード例 #13
0
ファイル: test_json.py プロジェクト: vitan/blaze
 def test_basic(self):
     dd = JSON(self.filename, 'r', dshape=self.dshape)
     self.assertRaises(Exception, lambda: tuple(dd))