SharkPython/shark-python/tests/TestMongoDB.py at master · SharedKnowledge/SharkPython · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
import unittest
import json
import jsonpickle
import bson
from pymongo import MongoClient

from Knowledge import Knowledge
from KnowledgeBase import KnowledgeBase
from PeerSemanticTag import Address, PeerSemanticTag
from PredicateSemanticTag import PredicateSemanticTag
from SemanticNet import SemanticNet
from SemanticTag import *
from Vocabulary import Vocabulary
from MongoHelper import *


class TestMongoDB(unittest.TestCase):


    #client = MongoClient('localhost', 27017)

    def test_CRUD(self):
        client = MongoClient()
        db = client.local
        property_tag = SemanticTag("Property", "ssn:Poperty")
        #property_tag_json = json.dumps({'name':property_tag.name,
           #                             'id':property_tag.id, 'si':property_tag.si})
        collection = db.semantic_tags
        #a = type(property_tag_json)
        #property_tag_id = collection.insert_one(json.dumps({'name':property_tag.name})).inserted_id
        collection.insert_one(        {
            "id": str(property_tag.id),
            "name": property_tag.name,

        })
        collection.insert_one(        {
            "id": 10,
            "name": "example",

        })
        all = db.semantic_tags.find()
        cursor = db.semantic_tags.find_one( {
            "name": property_tag.name
        })

        db.semantic_tags.delete_one({"name": property_tag.name})
        after_delete = db.semantic_tags.find()

        db.semantic_tags.update_one({"name": "example"}, { "$set": {"name": "example_after_update"}})

        after_update = db.semantic_tags.find_one({"name": "example_after_update"})
        property_tag.id = 10


    def test_semantic_tag_CRUD(self):
        property_tag = SemanticTag("Property", "ssn:Poperty")
        property_tag_json = jsonpickle.encode(property_tag)
        client = MongoClient()
        db = client.local
        collection = db.semantic_tags
        #print(type(property_tag_json))
        collection.insert_one(property_tag.__dict__)
        print(property_tag.__dict__)
        i =10
        property_tag_json_returned = db.semantic_tags.find_one( {
            "name": property_tag.name
        })
        #print(property_tag_json_returned["name"])
        #testList = property_tag_json_returned["si"]
        #print(property_tag_json_returned.keys())

    def test_knowledge_CRUD(self):
        tag_id = 0
        property_tag = SemanticTag("Property", "ssn:Poperty")
        measurement_property_tag = SemanticTag("Measurement Property", "ssn:property/MeasurementProperty")
        relation_tag1 = PredicateSemanticTag("is a", "http://www.w3.org/ns/ssn/#is_a", property_tag, measurement_property_tag)
        semantic_net = SemanticNet()
        semantic_net.add_tag(property_tag, tag_id)
        tag_id += 1
        semantic_net.add_tag(measurement_property_tag, tag_id)
        tag_id += 1
        semantic_net.add_predicate(relation_tag1)
        accuracy_tag = SemanticTag("Accuracy", "ssn:property/MeasurementProperty/Accuracy")
        relation_tag2 = PredicateSemanticTag("is a", "http://www.w3.org/ns/ssn/#is_a", measurement_property_tag, accuracy_tag)
        semantic_net.add_tag(accuracy_tag, tag_id)
        semantic_net.add_predicate(relation_tag2)
        vocabulary = Vocabulary(semantic_net, None, None, None, None)
        knowledge = Knowledge(vocabulary, None, None)
        kb = KnowledgeBase(knowledge)
        client = MongoClient()
        db = client.local
        collection = db.knowledge_base
        knowledge_json = jsonpickle.encode(knowledge)
        print("\n" + knowledge_json + "\n")
        print(type(knowledge_json))

        knowledge_bson = bson.son.SON(json.loads(knowledge_json))
        print(type(knowledge_bson))

        semantic_net_json = jsonpickle.encode(semantic_net)
        collection.insert_one(knowledge_bson)

        knowledge_found = collection.find_one(bson.son.SON(json.loads(knowledge_json)))

        print("Knowledge retrieved: " + str(knowledge_found))


    def test_mongo_helper(self):

        tag_id = 0
        property_tag = SemanticTag("Property", "ssn:Poperty")
        frequency_tag = SemanticTag("Frequency", "ssn:property/Frequency")
        relation_tag1 = PredicateSemanticTag("is a", "http://www.w3.org/ns/ssn/#is_a", property_tag, frequency_tag)
        semantic_net = SemanticNet()
        semantic_net.add_tag(property_tag, tag_id)
        tag_id += 1
        semantic_net.add_tag(frequency_tag, tag_id)
        tag_id += 1
        semantic_net.add_predicate(relation_tag1)
        precision_tag = SemanticTag("Accuracy", "ssn:property/MeasurementProperty/Precision")
        relation_tag2 = PredicateSemanticTag("is a", "http://www.w3.org/ns/ssn/#is_a", frequency_tag, precision_tag)
        semantic_net.add_tag(precision_tag, tag_id)
        semantic_net.add_predicate(relation_tag2)
        vocabulary = Vocabulary(semantic_net, None, None, None, None)
        knowledge = Knowledge(vocabulary, None, None)
        kb = KnowledgeBase(knowledge)
        client = MongoClient()
        db = client.local
        collection = db.knowledge_base

        MongoHelper.save_knowledge(knowledge)


        knowledge_json = jsonpickle.encode(knowledge)

        knowledge_found = MongoHelper.find_knowledge(bson.son.SON(json.loads(knowledge_json)))

        print("Knowledge retrieved: " + str(knowledge_found))