class VideoFilterTest(TestCase): def load_data(self, file_name): return json.load(file_name) def setUp(self): self._filter = Filter() def test_filter_youtube(self): self.print_json("resources/test/statuses.json") statuses = self.load_data("resources/test/statuses.txt") expected_content = MyContent() expected_content.content = "test" expected_content.user = 576279899 contents = self._filter.filter(statuses) self.asserTrue(expected_content in contents) def print_json(self, file_name): with open(file_name, "r") as file: statuses = self.load_data(file) pp = pprint.PrettyPrinter(indent=4) pp.pprint(statuses)
def setUp(self): self._filter = Filter()