def setUp(self):
self.subreddit1 = SubredditClass('Food1')
self.subreddit2 = SubredditClass('Food2')
self.subreddit3 = SubredditClass('Food3')
self.Food_Nov = open('Food-Nov.json', 'r+').read()
self.Food_Dec = open('Food-Dec.json', 'r+').read()
self.Food_Jan = open('Food-Jan.json', 'r+').read()
self.food1 = json.loads(self.Food_Nov)
self.food2 = json.loads(self.Food_Dec)
self.food3 = json.loads(self.Food_Jan)
DataParse.json_data_extract(self.food1, self.subreddit1, 'Food1')
DataParse.json_data_extract(self.food2, self.subreddit2, 'Food2')
DataParse.json_data_extract(self.food3, self.subreddit3, 'Food3')
def test_json_data_extract(self):
DataParse.json_data_extract(self.animals1, self.Animals, 'Animals')
DataParse.json_data_extract(self.animals2, self.Animals, 'Animals')
DataParse.json_data_extract(self.animals3, self.Animals, 'Animals')
DataParse.json_data_extract(self.food1, self.Food, 'Food')
DataParse.json_data_extract(self.food2, self.Food, 'Food')
DataParse.json_data_extract(self.food3, self.Food, 'Food')
assert len(self.Food.posts) == 30
assert self.Food.posts[0].getClass() == 'Food'
assert self.Food.posts[29].getClass() == 'Food'
assert self.Food.posts[0].getTitle() == 'banana'
assert self.Food.posts[29].getTitle() == 'cracker'
assert self.Food.posts[0].selftext == 'apple'
assert self.Food.posts[29].selftext == 'apple'
assert len(self.Animals.posts) == 30
assert self.Animals.posts[0].getClass() == 'Animals'
assert self.Animals.posts[29].getClass() == 'Animals'
assert self.Animals.posts[0].getTitle() == 'cat'
assert self.Animals.posts[29].getTitle() == 'octopus'
assert self.Animals.posts[0].selftext == 'dog'
assert self.Animals.posts[29].selftext == 'dog'
def setUp(self):
self.Animals_Nov = open('Animals-Nov.json', 'r+').read()
self.Animals_Dec = open('Animals-Dec.json', 'r+').read()
self.Animals_Jan = open('Animals-Jan.json', 'r+').read()
self.Food_Nov = open('Food-Nov.json', 'r+').read()
self.Food_Dec = open('Food-Dec.json', 'r+').read()
self.Food_Jan = open('Food-Jan.json', 'r+').read()
self.animals1 = json.loads(self.Animals_Nov)
self.animals2 = json.loads(self.Animals_Dec)
self.animals3 = json.loads(self.Animals_Jan)
self.food1 = json.loads(self.Food_Nov)
self.food2 = json.loads(self.Food_Dec)
self.food3 = json.loads(self.Food_Jan)
self.Animals = SubredditClass('Animals')
self.Food = SubredditClass('Food')
self.AnimalsTestClass = SubredditClass('Animals')
self.FoodTestClass = SubredditClass('Food')
DataParse.json_data_extract(self.animals1, self.Animals, 'Animals')
DataParse.json_data_extract(self.animals2, self.Animals, 'Animals')
DataParse.json_data_extract(self.animals3, self.AnimalsTestClass,
'Animals')
DataParse.json_data_extract(self.food1, self.Food, 'Food')
DataParse.json_data_extract(self.food2, self.Food, 'Food')
DataParse.json_data_extract(self.food3, self.FoodTestClass, 'Food')
self.Food.splitData('A')
self.Food.updateWords()
self.Animals.splitData('A')
self.Animals.updateWords()
self.FoodTestClass.splitData('T')
self.AnimalsTestClass.splitData('T')
self.FoodControl = self.Food.copy()
self.AnimalControl = self.Animals.copy()
self.Classifier = StatisticSet(self.Food, self.Animals)
