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)