Python KRWWatertype.code Examples

Example #1

File: update_db_for_suitable_measures_tests.py Project: lizardsystem/lizard-measure

    def test_g(self):
        """Test a watertype references the new WatertypeGroup."""
        watertype = KRWWatertype()
        watertype.code = 'O1'
        watertype.save()

        WatertypeGroups().create('K&O')

        watertype = KRWWatertype.objects.get(code='O1')
        self.assertEqual(WatertypeGroup.objects.get(code='K&O'), watertype.watertype_group)

Example #2

File: update_db_for_suitable_measures_tests.py Project: lizardsystem/lizard-measure

    def test_c(self):
        """Test a watertype does not reference the new WatertypeGroup."""
        watertype = KRWWatertype()
        watertype.code = 'R1'
        watertype.save()

        WatertypeGroups().create('M')

        watertype = KRWWatertype.objects.get(code='R1')
        self.assertEqual(None, watertype.watertype_group)