def test_compute_does_not_unload_from_memory(self): storage = StorageFactory().get_storage('dict_storage') storage.write_table(table_name='tests', table_data={ "a_dependent_variable": array([1, 5, 1000]), "id": array([1, 3, 4]) }) dataset = Dataset(in_storage=storage, in_table_name='tests', id_name="id", dataset_name="tests") values = dataset.get_attribute("a_dependent_variable") self.assert_( "a_dependent_variable" in dataset.get_attributes_in_memory()) dataset.compute_variables("opus_core.tests.a_test_variable") self.assert_( "a_dependent_variable" in dataset.get_attributes_in_memory()) self.assert_("a_test_variable" in dataset.get_attributes_in_memory()) # The type of values will be int32 on a 32-bit machine, and int64 on a 64 bit machine if platform.architecture()[0] == '64bit': self.assertEqual(values.dtype.type, int64) else: self.assertEqual(values.dtype.type, int32)
def test_compute_unloads_from_memory(self): storage = StorageFactory().get_storage('dict_storage') storage.write_table( table_name='tests', table_data={ 'a_dependent_variable':array([1,5,10]), 'id':array([1,3,4]) } ) dataset = Dataset(in_storage=storage, in_table_name='tests', id_name='id', dataset_name='tests') SessionConfiguration(in_storage=storage)["flush_variables"] = True dataset.get_attribute("a_dependent_variable") self.assert_("a_dependent_variable" in dataset.get_attributes_in_memory()) dataset.compute_variables("opus_core.tests.a_test_variable") self.assert_("a_dependent_variable" not in dataset.get_attributes_in_memory()) self.assert_("a_test_variable" in dataset.get_attributes_in_memory()) SimulationState().remove_singleton(delete_cache=True)
def test_compute_does_not_unload_from_memory(self): storage = StorageFactory().get_storage('dict_storage') storage.write_table( table_name='tests', table_data={ "a_dependent_variable":array([1,5,1000]), "id":array([1,3,4]) } ) dataset = Dataset(in_storage=storage, in_table_name='tests', id_name="id", dataset_name="tests") values = dataset.get_attribute("a_dependent_variable") self.assert_("a_dependent_variable" in dataset.get_attributes_in_memory()) dataset.compute_variables("opus_core.tests.a_test_variable") self.assert_("a_dependent_variable" in dataset.get_attributes_in_memory()) self.assert_("a_test_variable" in dataset.get_attributes_in_memory()) # The type of values will be int32 on a 32-bit machine, and int64 on a 64 bit machine if platform.architecture()[0]=='64bit': self.assertEqual(values.dtype.type, int64) else: self.assertEqual(values.dtype.type, int32)
def test_compute_unloads_from_memory(self): storage = StorageFactory().get_storage('dict_storage') storage.write_table(table_name='tests', table_data={ 'a_dependent_variable': array([1, 5, 10]), 'id': array([1, 3, 4]) }) dataset = Dataset(in_storage=storage, in_table_name='tests', id_name='id', dataset_name='tests') SessionConfiguration(in_storage=storage)["flush_variables"] = True dataset.get_attribute("a_dependent_variable") self.assert_( "a_dependent_variable" in dataset.get_attributes_in_memory()) dataset.compute_variables("opus_core.tests.a_test_variable") self.assert_( "a_dependent_variable" not in dataset.get_attributes_in_memory()) self.assert_("a_test_variable" in dataset.get_attributes_in_memory()) SimulationState().remove_singleton(delete_cache=True)