@codecademy/gamut-tests

Index: package/dist/index.d.ts
===================================================================
--- package/dist/index.d.ts
+++ package/dist/index.d.ts
@@ -1,7 +1,22 @@
-import { setupRtl as setupRtlBase } from 'component-test-setup';
+import { FullProps, RemainingPropsAndTestOverrides, RenderRtl, SetupComponentType } from 'component-test-setup';
 import * as React from 'react';
 export declare const MockGamutProvider: React.FC<{
     children?: React.ReactNode;
     useLogicalProperties?: boolean;
 }>;
-export declare const setupRtl: typeof setupRtlBase;
+type StripDataAttrs<T> = {
+    [K in keyof T as K extends `data-${string}` ? never : K]: T[K];
+};
+type RequiredKeys<T> = {
+    [K in keyof T]-?: Record<string, unknown> extends {
+        [P in K]: T[K];
+    } ? never : K;
+}[keyof T];
+type TestProps<Component extends SetupComponentType, BaseProps extends Partial<FullProps<Component>>> = RequiredKeys<StripDataAttrs<Omit<FullProps<Component>, keyof BaseProps>>> extends never ? [RemainingPropsAndTestOverrides<Component, BaseProps>?] : [RemainingPropsAndTestOverrides<Component, BaseProps>];
+export type GamutRenderRtlReturn<Component extends SetupComponentType, BaseProps extends Partial<FullProps<Component>>> = ReturnType<RenderRtl<Component, BaseProps>>;
+export type GamutRenderRtl<Component extends SetupComponentType, BaseProps extends Partial<FullProps<Component>>> = {
+    (...testProps: TestProps<Component, BaseProps>): GamutRenderRtlReturn<Component, BaseProps>;
+    options: (options: Parameters<RenderRtl<Component, BaseProps>['options']>[0]) => GamutRenderRtl<Component, BaseProps>;
+};
+export declare const setupRtl: <Component extends SetupComponentType, BaseProps extends Partial<FullProps<Component>> = {}>(Component: Component, baseProps?: BaseProps) => GamutRenderRtl<Component, BaseProps>;
+export {};