def testUnconsistantTypes(self): model = hdf5.Hdf5TreeModel() h5 = commonh5.File("/foo/bar/1.mock", "w") h5.create_group("aaa100") h5.create_group("100aaa") model.insertH5pyObject(h5) proxy = hdf5.NexusSortFilterProxyModel() proxy.setSourceModel(model) proxy.sort(0, qt.Qt.AscendingOrder) names = self.getChildNames(proxy, proxy.index(0, 0, qt.QModelIndex())) self.assertListEqual(names, ["100aaa", "aaa100"])
def testNumber(self): model = hdf5.Hdf5TreeModel() h5 = _mock.File("/foo/bar/1.mock") h5.create_group("a1") h5.create_group("a20") h5.create_group("a3") model.insertH5pyObject(h5) proxy = hdf5.NexusSortFilterProxyModel() proxy.setSourceModel(model) proxy.sort(0, qt.Qt.AscendingOrder) names = self.getChildNames(proxy, proxy.index(0, 0, qt.QModelIndex())) self.assertListEqual(names, ["a1", "a3", "a20"])
def testEndTimeInArray(self): model = hdf5.Hdf5TreeModel() h5 = _mock.File("/foo/bar/1.mock") h5.create_group("a").create_dataset("end_time", numpy.array([numpy.string_("2015")])) h5.create_group("b").create_dataset("end_time", numpy.array([numpy.string_("2013")])) h5.create_group("c").create_dataset("end_time", numpy.array([numpy.string_("2014")])) model.insertH5pyObject(h5) proxy = hdf5.NexusSortFilterProxyModel() proxy.setSourceModel(model) proxy.sort(0, qt.Qt.DescendingOrder) names = self.getChildNames(proxy, proxy.index(0, 0, qt.QModelIndex())) self.assertListEqual(names, ["a", "c", "b"])
def testNXentryName(self): """Test NXentry without start_time or end_time""" model = hdf5.Hdf5TreeModel() h5 = commonh5.File("/foo/bar/1.mock", "w") create_NXentry(h5, "a") create_NXentry(h5, "c") create_NXentry(h5, "b") model.insertH5pyObject(h5) proxy = hdf5.NexusSortFilterProxyModel() proxy.setSourceModel(model) proxy.sort(0, qt.Qt.AscendingOrder) names = self.getChildNames(proxy, proxy.index(0, 0, qt.QModelIndex())) self.assertListEqual(names, ["a", "b", "c"])
def testStartTime(self): """If it is not NXentry, start_time is not used""" model = hdf5.Hdf5TreeModel() h5 = commonh5.File("/foo/bar/1.mock", "w") h5.create_group("a").create_dataset("start_time", data=numpy.string_("2015")) h5.create_group("b").create_dataset("start_time", data=numpy.string_("2013")) h5.create_group("c").create_dataset("start_time", data=numpy.string_("2014")) model.insertH5pyObject(h5) proxy = hdf5.NexusSortFilterProxyModel() proxy.setSourceModel(model) proxy.sort(0, qt.Qt.AscendingOrder) names = self.getChildNames(proxy, proxy.index(0, 0, qt.QModelIndex())) self.assertListEqual(names, ["a", "b", "c"])
def testNXentryEndTimeInArray(self): """Test NXentry with end_time""" model = hdf5.Hdf5TreeModel() h5 = commonh5.File("/foo/bar/1.mock", "w") create_NXentry(h5, "a").create_dataset("end_time", data=numpy.array([numpy.string_("2015")])) create_NXentry(h5, "b").create_dataset("end_time", data=numpy.array([numpy.string_("2013")])) create_NXentry(h5, "c").create_dataset("end_time", data=numpy.array([numpy.string_("2014")])) model.insertH5pyObject(h5) proxy = hdf5.NexusSortFilterProxyModel() proxy.setSourceModel(model) proxy.sort(0, qt.Qt.DescendingOrder) names = self.getChildNames(proxy, proxy.index(0, 0, qt.QModelIndex())) self.assertListEqual(names, ["a", "c", "b"])