def test_default_stat_list(self): my_dict = {'Tech': 1, 'Contract': 4, 'Artifact': 3} stat = Stats(my_dict) self.assertEqual(stat.list_stats()[0]['stat'], 'Contract') self.assertEqual(len(stat.list_stats()), 3)
def test_filtered_stat_list(self): my_dict = {'Tech': 1, 'Contract': 4, 'Artifact': 3} stat = Stats(my_dict) self.assertEqual( stat.list_stats(['Tech', 'Artifact'])[0]['stat'], 'Artifact')