def test_unpickle_dict(self):
expected = {
1: {
"1ID": "A23",
"Gender": "Male",
"Age": 22,
"Sales": 2445,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
2: {
"IhD": "A2f3",
"Gender": "Male",
"Age": 23,
"Sales": 2565,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
3: {
"IjD": "Aa23",
"Gender": "Female",
"Age": 25,
"Sales": 25,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
4: {
"IgD": "A23",
"Gender": "Female",
"Age": 26,
"Sales": 225,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
}
}
pickle = Pickler.pickle_dictionary_values(expected)
self.local.insert_dictionary(pickle)
result = Unpickler.unpickle_dictionary(self.local.get_db())
self.assertEqual(expected, result)
def insert_remote_dict(self, dictionary):
"""Insert values into both the local and remote"""
pickled = Pickler.pickle_dictionary_values(dictionary)
self.remote.insert_dictionary(pickled)
class TestPicklerSetUp(TestCase):
# Wesley
def setUp(self):
self.pickler = Pickler()
# Wesley
def tearDown(self):
# self.pickler.dispose()
self.pickler = None
# Wesley
# def test_pickle_record_byte_stream(self):
# """Check if values are a byte stream when using pickle_record_values function"""
# key = "test"
# expected = bytes
# value = {"ID": "A23", "Gender": "Male", "Age": 22, "Sales": 245, "BMI": "normal", "salary": 20, "Birthday":
# "24/06/1995"}
# data = self.pickler.pickle_record_values(key, value)
# result = type(data["test"])
# self.assertEqual(result, expected)
# Wesley
# def test_pickle_record_return_dict(self):
# """Function returns a dictionary"""
# key = "test"
# expected = dict
# value = {"ID": "A23", "Gender": "Male", "Age": 22, "Sales": 245, "BMI": "normal", "salary": 20, "Birthday":
# "24/06/1995"}
# result = self.pickler.pickle_dictionary_values(key, value)
# self.assertIsInstance(result, expected)
# Wesley
def test_pickle_dictionary_type_byte(self):
"""True if all values in dictionary are of type 'byte'"""
expected = bytes
data = {
0: {
"1ID": "A23",
"Gender": "Male",
"Age": 22,
"Sales": 245,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
1: {
"IhD": "A2f3",
"Gender": "Male",
"Age": 22,
"Sales": 245,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
2: {
"IjD": "Aa23",
"Genkder": "Male",
"Age": 22,
"Sales": 245,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
3: {
"IgD": "A23",
"Gender": "Male",
"Age": 22,
"Sales": 245,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
}
}
data = self.pickler.pickle_dictionary_values(data)
result = (type(value) == expected for value in data.values())
self.assertTrue(all(result))
# Wesley
def test_pickle_dictionary_type_string(self):
"""False if any values in dictionary are of type 'string'"""
the_type = bytes
data = {
0: {
"1ID": "A23",
"Gender": "Male",
"Age": 22,
"Sales": 245,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
1: {
"IhD": "A2f3",
"Gender": "Male",
"Age": 22,
"Sales": 245,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
2: {
"IjD": "Aa23",
"Genkder": "Male",
"Age": 22,
"Sales": 245,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
3: {
"IgD": "A23",
"Gender": "Male",
"Age": 22,
"Sales": 245,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
}
}
data = self.pickler.pickle_dictionary_values(data)
data[2] = "This is a string"
result = (type(value) == the_type for value in data.values())
self.assertFalse(all(result))
# Wesley
def test_pickle_dictionary_type_string_true(self):
"""True if a value in dictionary are of type 'string'"""
the_type = str
data = {
0: {
"1ID": "A23",
"Gender": "Male",
"Age": 22,
"Sales": 245,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
1: {
"IhD": "A2f3",
"Gender": "Male",
"Age": 22,
"Sales": 245,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
2: {
"IjD": "Aa23",
"Genkder": "Male",
"Age": 22,
"Sales": 245,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
3: {
"IgD": "A23",
"Gender": "Male",
"Age": 22,
"Sales": 245,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
}
}
data = self.pickler.pickle_dictionary_values(data)
data[2] = "This is a string"
result = (type(value) == the_type for value in data.values())
self.assertTrue(any(result))
