def setUp(self): # Setup other related objects county = County.objects.get(name="Decatur") acc = Accusation(**accusation1) acc.save() race = Race(**race1) race.save() #Lynching self.lynching = Lynching(pca_id="22394") self.lynching.save() # Named Victim self.victim1 = Victim(**named_victim) self.victim1.save() self.victim1.county = county self.victim1.accusation.add(acc) self.victim1.race = race self.victim1.lynching = self.lynching self.victim1.save() # Unnamed Victim self.victim2 = Victim(**unnamed_victim) self.victim2.save() self.victim2.county = county self.victim2.accusation.add(acc) self.victim2.race = race self.victim2.lynching = self.lynching self.victim2.save()
class AccusationTest(TestCase): def setUp(self): self.acc = Accusation(label=accusation1["label"]) self.acc.save() def test_string(self): expected = "Test Crime" self.assertEqual(expected, "%s" % self.acc)
def _handle_accusation(self, victim, accusation_raw): """ Tries to approximate a match of the accusation from the current values or creates one if no match found. """ if not accusation_raw: return None accusation_text = accusation_raw.strip(' \t\n\r') try: accusation = Accusation.objects.get(label__iexact=accusation_text) except Accusation.DoesNotExist: accusation = Accusation(label=accusation_text) accusation.save() victim.accusation.add(accusation)