X-Git-Url: https://git.sesse.net/?p=movit;a=blobdiff_plain;f=white_balance_effect_test.cpp;h=4cd2ae0424af342fc56ecab6bf2a4e547e25b483;hp=966cba3d9b28dabd7d25ab82a927a9dec60c65ab;hb=7d78534e38d50892ca57d3a17f27214ac094ea58;hpb=b808921f84d52f426bc7bb22480f0806668a45a1

diff --git a/white_balance_effect_test.cpp b/white_balance_effect_test.cpp
index 966cba3..4cd2ae0 100644
--- a/white_balance_effect_test.cpp
+++ b/white_balance_effect_test.cpp
@@ -1,7 +1,11 @@
 // Unit tests for WhiteBalanceEffect.
 
-#include "test_util.h"
+#include <iomanip>
+
+#include "effect_chain.h"
 #include "gtest/gtest.h"
+#include "image_format.h"
+#include "test_util.h"
 #include "white_balance_effect.h"
 
 TEST(WhiteBalanceEffectTest, GrayNeutralDoesNothing) {
@@ -17,7 +21,7 @@ TEST(WhiteBalanceEffectTest, GrayNeutralDoesNothing) {
 	};
 
 	float out_data[5 * 4];
-	EffectChainTester tester(data, 1, 5, FORMAT_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
+	EffectChainTester tester(data, 1, 5, FORMAT_RGBA_POSTMULTIPLIED_ALPHA, COLORSPACE_sRGB, GAMMA_LINEAR);
 	Effect *white_balance_effect = tester.get_chain()->add_effect(new WhiteBalanceEffect());
 	ASSERT_TRUE(white_balance_effect->set_vec3("neutral_color", neutral));
 	tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
@@ -36,7 +40,7 @@ TEST(WhiteBalanceEffectTest, SettingReddishNeutralColorNeutralizesReddishColor)
 	};
 
 	float out_data[3 * 4];
-	EffectChainTester tester(data, 1, 3, FORMAT_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
+	EffectChainTester tester(data, 1, 3, FORMAT_RGBA_POSTMULTIPLIED_ALPHA, COLORSPACE_sRGB, GAMMA_LINEAR);
 	Effect *white_balance_effect = tester.get_chain()->add_effect(new WhiteBalanceEffect());
 	ASSERT_TRUE(white_balance_effect->set_vec3("neutral_color", neutral));
 	tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
@@ -73,7 +77,7 @@ TEST(WhiteBalanceEffectTest, HigherColorTemperatureIncreasesBlue) {
 	};
 
 	float out_data[2 * 4];
-	EffectChainTester tester(data, 1, 2, FORMAT_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);
+	EffectChainTester tester(data, 1, 2, FORMAT_RGBA_POSTMULTIPLIED_ALPHA, COLORSPACE_sRGB, GAMMA_LINEAR);
 	Effect *white_balance_effect = tester.get_chain()->add_effect(new WhiteBalanceEffect());
 	ASSERT_TRUE(white_balance_effect->set_float("output_color_temperature", 10000.0f));
 	tester.run(out_data, GL_RGBA, COLORSPACE_sRGB, GAMMA_LINEAR);