Update test_data_compute.py