def test_remove_by_type(self): RecordPrototype.create(type=relations.RECORD_TYPE.TEST_INT, date=self.date, value_int=41.7) record = RecordPrototype.create(type=relations.RECORD_TYPE.TEST_FLOAT, date=self.date, value_float=66.6) RecordPrototype.remove_by_type(relations.RECORD_TYPE.TEST_INT) self.assertEqual(list(RecordPrototype._db_all().values_list('id', flat=True)), [record.id])