Python Experiment Examples

Example #1

    def test_get_metadata_key_ranges(self):
        experiment = m.Experiment(
            os.path.join(ud.UData().Basedir, "foo.myrmidon"))
        experiment.SetMetaDataKey("alive", True)
        experiment.CreateAnt()
        a = experiment.CreateAnt()
        a.SetValue(key="alive", value=False, time=m.Time())
        a = experiment.CreateAnt()
        a.SetValue(key="alive", value=False, time=m.Time.SinceEver())
        a.SetValue(key="alive", value=True, time=m.Time())
        a.SetValue(key="alive", value=True, time=m.Time().Add(1))
        a.SetValue(key="alive", value=False, time=m.Time().Add(2))
        a.SetValue(key="alive", value=True, time=m.Time().Add(3))

        ranges = m.Query.GetMetaDataKeyRanges(experiment,
                                              key="alive",
                                              value=True)
        self.assertEqual(len(ranges), 4)
        self.assertEqual(ranges[0], (1, m.Time.SinceEver(), m.Time.Forever()))
        self.assertEqual(ranges[1], (2, m.Time.SinceEver(), m.Time()))
        self.assertEqual(ranges[2], (3, m.Time(), m.Time().Add(2)))
        self.assertEqual(ranges[3], (3, m.Time().Add(3), m.Time.Forever()))

        with self.assertRaises(IndexError):
            m.Query.GetMetaDataKeyRanges(experiment, key="isDead", value=True)

        with self.assertRaises(ValueError):
            m.Query.GetMetaDataKeyRanges(experiment, key="alive", value=42.0)

Example #2

    def test_identification_manipulation(self):
        ants = [self.experiment.CreateAnt(), self.experiment.CreateAnt()]
        with self.assertRaises(IndexError):
            self.experiment.AddIdentification(42, 0, m.Time.SinceEver(),
                                              m.Time.Forever())
        self.experiment.AddIdentification(ants[0].ID, 0, m.Time.SinceEver(),
                                          m.Time.Forever())

        with self.assertRaises(m.OverlappingIdentification):
            self.experiment.AddIdentification(ants[0].ID, 1,
                                              m.Time.SinceEver(),
                                              m.Time.Forever())

        with self.assertRaises(m.OverlappingIdentification):
            self.experiment.AddIdentification(ants[1].ID, 0,
                                              m.Time.SinceEver(),
                                              m.Time.Forever())

        self.experiment.AddIdentification(ants[1].ID, 1, m.Time.SinceEver(),
                                          m.Time.Forever())

        e2 = m.Experiment("foo.myrmidon")
        a2 = e2.CreateAnt()
        i2 = e2.AddIdentification(a2.ID, 0, m.Time.SinceEver(),
                                  m.Time.Forever())

        with self.assertRaises(ValueError):
            self.experiment.DeleteIdentification(i2)

        ants[0].Identifications[0].End = m.Time()
        with self.assertRaises(RuntimeError):
            self.experiment.FreeIdentificationRangeAt(
                ants[0].Identifications[0].TagValue,
                m.Time().Add(-1))

        low, high = self.experiment.FreeIdentificationRangeAt(
            ants[0].Identifications[0].TagValue, m.Time())
        self.assertTimeEqual(low, m.Time())
        self.assertTimeEqual(high, m.Time.Forever())

        low, high = self.experiment.FreeIdentificationRangeAt(42, m.Time())
        self.assertTimeEqual(low, m.Time.SinceEver())
        self.assertTimeEqual(high, m.Time.Forever())

        identifications = self.experiment.IdentificationsAt(
            time=m.Time().Add(-1), removeUnidentifiedAnt=True)
        self.assertEqual(len(identifications), 2)
        self.assertEqual(identifications[ants[0].ID], 0)
        self.assertEqual(identifications[ants[1].ID], 1)
        identifications = self.experiment.IdentificationsAt(
            time=m.Time(), removeUnidentifiedAnt=True)
        self.assertEqual(len(identifications), 1)
        self.assertEqual(identifications[ants[1].ID], 1)
        self.assertFalse(ants[0].ID in identifications)
        identifications = self.experiment.IdentificationsAt(
            time=m.Time(), removeUnidentifiedAnt=False)
        self.assertEqual(len(identifications), 2)
        self.assertEqual(identifications[ants[0].ID], 2**32 - 1)
        self.assertEqual(identifications[ants[1].ID], 1)

Example #3

    def test_file_manipulation(self):
        dirs = [
            ud.UData().Basedir / "test-manipulation",
            ud.UData().Basedir / "test-manipulation-new"
        ]
        for d in dirs:
            try:
                os.makedirs(d)
            except FileExistsError:
                pass

        filepath = str(dirs[0] / "test.myrmidon")
        goodNewPath = str(dirs[0] / "test2.myrmidon")
        badNewPath = str(dirs[1] / "test.myrmidon")
        e = m.Experiment(filepath)
        e.Save(filepath)
        e.Save(goodNewPath)
        with self.assertRaises(ValueError):
            e.Save(badNewPath)

Example #4

File: test_ant.py Project: formicidae-tracker/myrmidon

 def setUp(self):
     self.experiment = m.Experiment("test-myrmidon")

Example #5

File: test-py-fort-myrmidon.py Project: formicidae-tracker/myrmidon

import py_fort_myrmidon

e = py_fort_myrmidon.Experiment('foo.myrmidon')
e.Name = 'foo'

Example #6

File: test_zone.py Project: formicidae-tracker/myrmidon

 def setUp(self):
     self.experiment = m.Experiment(
         str(ud.UData().Basedir / "zone-utests.myrmidon"))
     self.space = self.experiment.CreateSpace("nest")

Example #7

File: test_identification.py Project: formicidae-tracker/myrmidon

 def setUp(self):
     self.experiment = m.Experiment(
         str(ud.UData().Basedir / "identification-utest.myrmidon"))
     self.ant = self.experiment.CreateAnt()
     self.i = self.experiment.AddIdentification(self.ant.ID, 123)

Example #8

 def setUp(self):
     self.experiment = m.Experiment(
         str(ud.UData().Basedir / 'public-experiment.myrmidon'))