def test_drops_attrs_in_list(self): self.config[module] = {"attr_list": ["code"]} for d in self.data: nt.assert_true("code" in d) res = list(drop_attributes(self.config, self.data)) for r in res: nt.assert_true("code" not in r)
def test_works_when_data_is_a_generator(self): self.config[module] = {"attr_list": ["code", "token"]} gen_data = (d for d in self.data) res = list(drop_attributes(self.config, gen_data)) nt.assert_equals(len(res), 1)
def test_lines_are_deleted_if_empty_dicts(self): self.config[module] = {"attr_list": ["code", "token"]} res = list(drop_attributes(self.config, self.data)) nt.assert_equals(len(res), 1)
def test_drop_multiple_attrs(self): self.config[module] = {"attr_list": ["code", "token"]} res = list(drop_attributes(self.config, self.data)) for r in res: nt.assert_true("token" not in r) nt.assert_true("code" not in r)
def test_doesnt_crash_if_dict_doesnt_have_attr(self): self.config[module] = {"attr_list": ["test"]} res = list(drop_attributes(self.config, self.data)) for r in res: nt.assert_true("test" not in r)
def test_only_drops_attrs_in_list(self): self.config[module] = {"attr_list": ["code"]} res = list(drop_attributes(self.config, self.data)) for r in res: nt.assert_true("token" in r)
def test_doesnt_drop_attrs_by_default(self): res = list(drop_attributes(self.config, self.data)) nt.assert_equals(res, self.data)